poj 1273 Drainage Ditches

最大流入门题目

注意重边的处理

/*最大流问题*/
#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;


int n,m;
int c[205][205];
int flow[205][205];
int a[205];
int pre[205];
const int inf = 0x7fffffff;


int maxFlow()
{
    int ans = 0;
    memset(flow,0,sizeof(flow));
    queue<int>q;
    while(1)
    {
        memset(a,0,sizeof(a));
        a[1] = inf;
        q.push(1);
        while(!q.empty())
        {
            int temp = q.front();
            q.pop();
            for(int i = 1;i<=n;i++)
            {
                if(!a[i]&&c[temp][i]>flow[temp][i])
                {
                    q.push(i);
                    pre[i] = temp;
                   if(c[temp][i]-flow[temp][i]<a[temp])
                    a[i] = c[temp][i]-flow[temp][i];
                    else
                    a[i] = a[temp];
                }
            }
        }
        if(a[n]==0)
            break;
        for(int i = n;i!=1;i=pre[i])
        {
            flow[pre[i]][i] += a[n];
            flow[i][pre[i]] -= a[n];
        }
        ans += a[n];
    }
    return ans;
}


int main()
{
    int u,v,w;
    while(scanf("%d %d",&m,&n)!=EOF)
    {
        memset(c,0,sizeof(c));
        for(int i=0;i<m;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            c[u][v] += w;
        }
        int ans = maxFlow();
        printf("%d\n",ans);
    }
    return 0;
}