Codevs 1993 草地排水

1993 草地排水

时间限制: 2 s　　　　空间限制: 256000 KB　　　　题目等级 : 钻石 Diamond

题目描述 Description

在农夫约翰的农场上，每逢下雨，Bessie最喜欢的三叶草地就积聚了一潭水。这意味着草地被水淹没了，并且小草要继续生长还要花相当长一段时间。因此，农夫约翰修建了一套排水系统来使贝茜的草地免除被大水淹没的烦恼（不用担心，雨水会流向附近的一条小溪）。作为一名一流的技师，农夫约翰已经在每条排水沟的一端安上了控制器，这样他可以控制流入排水沟的水流量。

农夫约翰知道每一条排水沟每分钟可以流过的水量，和排水系统的准确布局（起点为水潭而终点为小溪的一张网）。需要注意的是，有些时候从一处到另一处不只有一条排水沟。

根据这些信息，计算从水潭排水到小溪的最大流量。对于给出的每条排水沟，雨水只能沿着一个方向流动，注意可能会出现雨水环形流动的情形。

输入描述 Input Description

第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 Description

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

样例输入 Sample Input

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

样例输出 Sample Output

50

#include<queue>
#include<vector>
#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
#define INF 0x7f
int n,m,map[205][205]={0},lays[205];
bool vis[205];
bool BFS(){
    queue<int>q;
    memset(lays,-1,sizeof(lays));
    lays[1]=0;q.push(1);
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=1;i<=m;i++){
            if(map[u][i]>0&&lays[i]==-1){
                lays[i]=lays[u]+1;
                if(i==m)return 1;
                else q.push(i);
            }
        }
    }
    return 0;
}
int Dinic(){
    int maxf=0;vector<int>st;
    while(BFS()){
        st.push_back(1);
        memset(vis,0,sizeof(vis));vis[1]=1;
        while(!st.empty()){
            int p=st.back();
            if(p==m){
                int minn,minx=INF;
                for(int i=1;i<st.size();i++){
                    int u=st[i-1],v=st[i];
                    if(map[u][v]>0&&map[u][v]<minx){
                        minn=u;minx=map[u][v];
                    }
                }
                maxf+=minx;
                for(int i=1;i<st.size();i++){
                    int u=st[i-1],v=st[i];
                    map[u][v]-=minx;
                    map[v][u]+=minx;
                }
                while(!st.empty()&&st.back()!=minn){
                    vis[st.back()]=0;st.pop_back();
                }
            }
            else {
                int i;
                for(i=1;i<=m;i++){
                    if(map[p][i]>0&&lays[i]==
                       lays[p]+1&&!vis[i]){
                            vis[i]=1;st.push_back(i);
                            break;
                       }
                }
                if(i>m)st.pop_back();
            }
        }
    }
    return maxf;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,a,b,c;i<=n;i++){
        scanf("%d%d%d",&a,&b,&c);
        map[a][b]+=c;
    }
    printf("%d\n",Dinic());
    return 0;
}