USACO4.2.1 网络流最大流算法

/*
ID:hk945801
TASK:ditch
LANG:C++
*/
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int inf=10000001;
struct node{
	int c,f;
}a[201][201];
int p[201*201],d[201*201],pre[201];
int main(){
	int i,j,k,m,n;
	int x;
	freopen("ditch.in","r",stdin);
	freopen("ditch.out","w",stdout);
	cin>>m>>n;  
    for(i=1;i<=m;i++){
	    int x,y,z;  
        scanf("%d%d%d",&x,&y,&z);                    
        a[x][y].c+=z;  
    } 
    int pd=1;
    while(pd){
	    pd=0;
	    int fi=0,l=1;
	    d[1]=1;
	    for(i=1;i<=n;i++)p[i]=0;  
        p[1]=1;  
        while(fi<l){  
            fi++;
        x=d[fi];
            for(i=2;i<=n;i++)
                if(!p[i] && (a[x][i].c>a[x][i].f || a[i][x].f>0)){  
                    d[++l]=i;
                    p[i]=1;
                    pre[i]=x;
                    if(i==n){pd=1;break;}  
                }         
            if(pd==1)break;  
        }  
        if(pd==0)break;  
        int min=inf;  
        x=n;  
        while(x!=1){  
            if(a[pre[x]][x].c>a[pre[x]][x].f&& a[pre[x]][x].c-a[pre[x]][x].f<min)  
                min=a[pre[x]][x].c-a[pre[x]][x].f;             
            else if(a[x][pre[x]].f>0 && a[x][pre[x]].f<min)  
                min=a[x][pre[x]].f;  
            x=pre[x];  
        }  
        x=n;  
        while(x!=1){  
            if(a[pre[x]][x].c>a[pre[x]][x].f)  
                a[pre[x]][x].f+=min;              
            else if(a[x][pre[x]].f>0)  
                a[x][pre[x]].f-=min;  
            x=pre[x];
        }
	}
	int ans=0;  
    for(i=2;i<=n;i++)  
        ans+=a[1][i].f;  
    cout<<ans<<endl;
	return 0;
}