HDU 1532 Drainage Ditches

HDU_1532

    这是一个比较基础的网络流的题目，好长时间没有写过网络流了，一开始把cap[v][u]赋成了-cap[u][v]，导致WA了好多次，看来是和最小费用最大流的cost[v][u]=-cost[u][v]搞混了。

#include<stdio.h>
#include<string.h>
int cap[210][210],flow[210][210];
int q[210],p[210],a[210];
int main()
{
    int i,j,k,N,M,u,v,front,rear;
    long long int ans;
    while(scanf("%d%d",&N,&M)==2)
    {
        memset(cap,0,sizeof(cap));
        for(i=0;i<N;i++)
        {
            scanf("%d%d%d",&u,&v,&k);
            cap[u][v]+=k;
        }
        memset(flow,0,sizeof(flow));
        ans=0;
        while(1)
        {
            memset(a,0,sizeof(a));
            a[1]=1000000000;
            front=rear=0;
            q[rear++]=1;
            while(front<rear)
            {
                u=q[front++];
                for(v=1;v<=M;v++)
                    if(!a[v]&&cap[u][v]>flow[u][v])
                    {
                        q[rear++]=v;
                        p[v]=u;
                        a[v]=a[u];
                        if(cap[u][v]-flow[u][v]<a[v])
                            a[v]=cap[u][v]-flow[u][v];
                    }
            }
            if(a[M]==0)
                break;
            for(v=M;v!=1;v=p[v])
            {
                flow[p[v]][v]+=a[M];
                flow[v][p[v]]-=a[M];
            }
            ans+=a[M];
        }
        printf("%I64d\n",ans);
    }
    return 0;    
}