poj 1459 最大流(EK实现)

Problem is from:http://poj.org/problem?id=1459

Edmond Karp最大流

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 110
int n,map[MAXN][MAXN],pre[MAXN];
bool flag[MAXN];

//BFS()寻找增广路
bool BFS(int start,int end){
    queue<int> que;
    int k;
    memset(pre,-1,sizeof(pre));
    memset(flag,true,sizeof(flag));
    que.push(start);
    flag[start]=false;
    while(!que.empty()){
        k=que.front();
        if(k==end)
            return true;
        que.pop();
        for(int i=0;i<n;i++){
            if(map[k][i] && flag[i]){
                flag[i]=false;
                pre[i]=k;
                que.push(i);
            }
        }
    }
    return false;
}

//MaxFlow()返回最大流
int MaxFlow(int start,int end){
    int ans=0,k,minf;
    while(BFS(start,end)){
        minf=-1;
        k=end;
        while(pre[k]!=-1){
            if(minf<0)
                minf=map[pre[k]][k];
            else
                minf=min(minf,map[pre[k]][k]);
            k=pre[k];
        }
        ans+=minf;
        k=end;
        while(pre[k]!=-1){
            map[pre[k]][k]-=minf;
            map[k][pre[k]]-=minf;
            k=pre[k];
        }
    }
    return ans;
}

int main()
{
    int np,nc,m,u,v,z,start,end;
    while(scanf("%d%d%d%d",&n,&np,&nc,&m)!=EOF){
        memset(map,0,sizeof(map));
        start=n;end=n+1;
        for(int i=0;i<m;i++){
            scanf(" (%d,%d)%d",&u,&v,&z);
            map[u][v]=z;
        }
        for(int i=0;i<np;i++){
            scanf(" (%d)%d",&v,&z);
            map[start][v]=z;
        }
        for(int i=0;i<nc;i++){
            scanf(" (%d)%d",&u,&z);
            map[u][end]=z;
        }
        n+=2;
        printf("%d\n",MaxFlow(start,end));
    }
    return 0;
}

直接套模版...