dinic+当前弧优化 模板

poj1273

#include<stdio.h>
#include<cstring>
#include<queue>
#define clear(a) memset(a,0,sizeof(a));
#define clearv(s) memset(s,false,sizeof(s));
using namespace std;
const int inf=0x3f3f3f3f;
int h[201],dis[201],m,n,num,flow,cur[201];
bool vis[201];
inline const int read(){
    register int f=1,x=0;
    register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return f*x;
}
struct edge{
    int v,nxt,c;
}e[4005];
void add(int u,int v,int w){
    num++;
    e[num].v=v;
    e[num].c=w;
    e[num].nxt=h[u];
    h[u]=num;
}
bool bfs(){
    memset(dis,-1,sizeof(dis));
    dis[1]=0;
    queue<int> q;
    q.push(1);
    while(!q.empty()){
         int u=q.front();
         q.pop();
         for(int i=h[u];i;i=e[i].nxt){
             int v=e[i].v;
             if(dis[v]==-1&&e[i].c>0){
                dis[v]=dis[u]+1;
                q.push(v);
             }
         }
    }
    if(dis[n]==-1) return false;
    return true;
}
int dfs(int u,int a){
    if(a<=0||u==n) return a;
    int f,flow=0,x;
    for(int i=h[u];i&&a;i=e[i].nxt){
        int v=e[i].v;
        if(dis[v]==dis[u]+1&&e[i].c){
           x=min(e[i].c,a);
           f=dfs(v,x);
           e[i].c-=f;
           e[i^1].c+=f;
           flow+=f;
           a-=f;
           if(e[i].c) cur[u]=i;
        }
    } 
    if(!flow) dis[u]=-1;
    return flow;
}
void init(){
    num=1,clear(h);
    flow=0;
}
int main(){
    while(~scanf("%d%d",&m,&n)){
    init();
    for (register int i=1;i<=m;i++){
        int x,y,z;
        x=read(),y=read(),z=read();
        add(x,y,z);
        add(y,x,0);
    }
    while(bfs()) {for(int i=1;i<=n;i++) cur[i]=h[i];flow+=dfs(1,inf);}
    printf("%d\n",flow);
    }
}