首页 > 代码库 > POJ 1273 Drainage Ditches(我的EK算法模板)

POJ 1273 Drainage Ditches(我的EK算法模板)

题意:给你n条边,目标位置t;接下来是每条边,包括起点,终点,容量;

感想:第一道最大流的代码,这道题是我更深地理解了Ek算法,不过这道题有个超坑的情况,那就是出现重边的情况==!

思路:EK算法

AC代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
#define INF 100000000
#define N 202
int cap[N][N],flow[N][N];
int p[N],a[N];
int n,t;
int Edmonds_Karp(int s)
{
    int f=0;
    queue<int >q;

    while(1)
    {
        memset(a,0,sizeof(a));
        a[s]=INF;
        q.push(s);
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            for(int v=1;v<=t;v++)
            if(!a[v]&&cap[u][v]>flow[u][v])
            {
                p[v]=u;
                q.push(v);
                a[v]=min(a[u],cap[u][v]-flow[u][v]);
            }
        }
        if(a[t]==0)break;
        for(int u=t;u!=s;u=p[u])
        {
            flow[p[u]][u]+=a[t];
            flow[u][p[u]]-=a[t];
        }
        f+=a[t];
    }
    return  f;

}
int main()
{
    int u,v,w,i,j;
    while(scanf("%d %d",&n,&t)!=EOF)
    {
        memset(cap,0,sizeof(cap));
        memset(flow,0,sizeof(flow));
        for(i=0;i<n;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            cap[u][v]+=w;
        }
        printf("%d\n",Edmonds_Karp(1));
    }

    return 0;
}