USACO 4.2 Drainage Ditches 草地排水(预流推进)

Description

　　在农夫约翰的农场上，每逢下雨，Bessie最喜欢的三叶草地就积聚了一潭水。这意味着草地被水淹没了，并且小草要继续生长还要花相当长一段时间。因此，农夫约翰修建了一套排水系统来使贝茜的草地免除被大水淹没的烦恼（不用担心，雨水会流向附近的一条小溪）。作为一名一流的技师，农夫约翰已经在每条排水沟的一端安上了控制器，这样他可以控制流入排水沟的水流量。
　　农夫约翰知道每一条排水沟每分钟可以流过的水量，和排水系统的准确布局（起点为水潭而终点为小溪的一张网）。需要注意的是，有些时候从一处到另一处不只有一条排水沟。
　　根据这些信息，计算从水潭排水到小溪的最大流量。对于给出的每条排水沟，雨水只能沿着一个方向流动，注意可能会出现雨水环形流动的情形。

Input

第1行: 两个用空格分开的整数N (0 <= N <= 200) 和 M (2 <= M <= 200)。N是农夫John已经挖好的排水沟的数量，M是排水沟交叉点的数量。交点1是水潭，交点M是小溪。
第二行到第N+1行: 每行有三个整数，Si, Ei, 和 Ci。Si 和 Ei (1 <= Si, Ei <= M) 指明排水沟两端的交点，雨水从Si 流向Ei。Ci (0 <= Ci <= 10,000,000)是这条排水沟的最大容量。

Output

输出一个整数，即排水的最大流量。

Sample Input

5 4
1 2 40
1 4 20
2 4 20
2 3 30
3 4 10

Sample Output

50

Solution

Dinic

#include
#include
#include
#include
#define N 205
#define INF 0x3f3f3f3f
using namespace std;
struct
{
    int to,c,p;
}e[N*2];
queue<int>q;
int n,m,c=1,last[N],d[N],s=1,t,cur[N],ans;
void add(int x,int y,int z)
{
    e[++c].to=y; e[c].c=z; e[c].p=last[x]; last[x]=c;
    e[++c].to=x; e[c].c=0; e[c].p=last[y]; last[y]=c;
}
void init() 
{
    scanf("%d%d",&n,&m);
    t=m;
    for(int i=1;i<=n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
    }
}
bool bfs()
{
    memset(d,0,sizeof(d));
    d[s]=1;
    while(!q.empty()) q.pop();
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=last[u];i;i=e[i].p)
            if(e[i].c&&!d[e[i].to])
            {
                d[e[i].to]=d[u]+1;
                if(e[i].to==t) return 1;
                q.push(e[i].to);
            }
    }
    return 0;
}
int dfs(int x,int maxf)
{
    if(x==t||!maxf) return maxf;
    int r=0;
    for(int &i=cur[x];i;i=e[i].p)
        if(e[i].c&&d[e[i].to]==d[x]+1)
        {
            int f=dfs(e[i].to,min(e[i].c,maxf-r));
            e[i].c-=f;
            e[i^1].c+=f;
            r+=f;
            if(r==maxf) break;
        } 
    return r;
}
void dinic() 
{
    while(bfs())
    {
        memcpy(cur,last,sizeof(cur));
        ans+=dfs(s,INF);
    }
}
int main()
{
    init();
    dinic();
    printf("%d",ans);
}