洛谷P3376【模板】网络最大流　 Dinic模板

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
typedef long long LL;
const int maxn=20050,maxm=200050,INF=0x7f7f7f7f;
using namespace std;
int n,m,s,t,u,v,w,ecnt=1,fir[maxn],dis[maxn],cur[maxn],ans;
struct edge {
    int from,to,cap,flow,nxt;
    edge(){}
    edge(int from,int to,int cap,int flow,int nxt):from(from),to(to),cap(cap),flow(flow),nxt(nxt){}
}e[maxm];
void add(int u,int v,int w) {
    e[++ecnt]=edge(u,v,w,0,fir[u]);     
    e[++ecnt]=edge(v,u,0,0,fir[v]); 
    fir[u]=ecnt-1; fir[v]=ecnt;
}
void init() {
    scanf("%d%d%d%d",&n,&m,&s,&t);
    for(int i=1;i<=m;i++) {
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
}
queue<int>que;
int bfs(int s,int t) {
    memset(dis,0,sizeof(dis));
    dis[s]=1; 
    que.push(s);
    while(!que.empty()) {
        int x=que.front() ;que.pop();
        for(int i=fir[x];i;i=e[i].nxt) 
        if(!dis[e[i].to]&&e[i].flow<e[i].cap) {
            dis[e[i].to]=dis[x]+1;
            que.push(e[i].to); 
        }
    }
    return dis[t];
}
int dfs(int x,int a) {
    if(x==t||a==0) return a;
    int fl,f=0;
    for(int &i=cur[x];i;i=e[i].nxt) {
    if(dis[e[i].to]==dis[x]+1&&(fl=dfs(e[i].to,min(a,e[i].cap-e[i].flow)))){
        e[i].flow+=fl;
        e[i^1].flow-=fl;
        a-=fl;
        f+=fl;
        }
        if(a==0) break;
    }
    return f;
}
int Dinic(int s,int t) {
    int res=0;
    while(bfs(s,t)) {
        for(int i=1;i<=n;i++) cur[i]=fir[i];
        res+=dfs(s,INF);
    }
    return res;
}
void work() {
    ans=Dinic(s,t);
    printf("%d\n",ans);
}
int main()
{
    init();
    work();
    return 0;
}