1. 程式人生 > 其它 >網路流 FLPP 板子

網路流 FLPP 板子

#include<bits/stdc++.h>
using namespace std;
const int MM=4e5+5,inf=0x3f3f3f3f;
int n,m,s,t,tot=1,a;
int nxt[MM],head[MM],to[MM],w[MM];
int h[MM],e[MM],gap[MM],in[MM];
struct cmp
{
    bool operator()(int a,int b) const 
    {
        return h[a]<h[b];
    } 
}; 
priority_queue<int,vector<int
>,cmp> q; void add(int u,int v,int flow) { nxt[++tot]=head[u]; to[tot]=v; w[tot]=flow; head[u]=tot; } bool bfs() { int q[1200],l=1,r=1,now; memset(h,0x3f,sizeof(h)); q[1]=t;h[t]=0; while(r>=l) { now=q[l++]; for(int i=head[now];i;i=nxt[i])
if(w[i^1]&&h[to[i]]>h[now]+1) h[to[i]]=h[now]+1,q[++r]=to[i]; } return h[s]!=inf; } void push(int now) { int flow; for(int i=head[now];i;i=nxt[i]) if(w[i]&&h[to[i]]+1==h[now]) { flow=min(e[now],w[i]); w[i]-=flow;w[i^1
]+=flow;e[now]-=flow,e[to[i]]+=flow; if(to[i]!=s&&to[i]!=t&&!in[to[i]]) q.push(to[i]),in[to[i]]=1; if(!e[now]) break; } } void relabel(int now) { h[now]=inf; for(int i=head[now];i;i=nxt[i]) if(w[i]&&h[to[i]]+1<h[now]) h[now]=h[to[i]]+1; return; } void pp() { for(int i=1;i<=n;i++) cout<<i<<' '; cout<<endl; for(int i=1;i<=n;i++) cout<<e[i]<<' '; cout<<endl; } int main() { cin>>n>>m>>s>>t; int u,v,a,now; for(int i=1;i<=m;i++) cin>>u>>v>>a,add(u,v,a),add(v,u,0); if(!bfs()) { cout<<0; return 0; } h[s]=n; for(int i=1;i<=n;i++) if(h[i]<inf) ++gap[h[i]]; for(int i=head[s];i;i=nxt[i]) { int flow=w[i]; w[i]-=flow;w[i^1]+=flow;e[s]-=flow;e[to[i]]+=flow; if(to[i]!=s&&to[i]!=t&&!in[to[i]]) q.push(to[i]),in[to[i]]=1; } while(!q.empty()) { now=q.top();q.pop();in[now]=0;push(now);//cout<<now<<endl; if(e[now]) { if(!--gap[h[now]]) for(int i=1;i<=n;i++) if(i!=s&&i!=t&&h[i]>h[now]&&h[i]<n+1) h[i]=n+1; relabel(now);++gap[h[now]]; q.push(now);in[now]=1; } } cout<<e[t]; return 0; } ~~~