POJ3436 ACM Computer Factory(dinic最大流+统计不同弧上流量的变化)

传送门：http://poj.org/problem?id=3436
这题的题意实在是麻烦，看题目看了好久好久。。。
题意：输入P和N，P表示零件数，N表示机器数。
0——必须没有
1——必须有
2——可有可无
机器要组装零件，每个机器有2*P+1个数字来描述，第一个数字是装配速度，可以理解为容量，然后的P个数字是输入的要求，后P个数字是输出结果，问你最大流和变化的流量。
思路很简单，没有什么需要思考的地方。
建图的时候建立一个源点和一个汇点。如果一个机器的输入没有1，就可以作为起点，就从源点跟他连边。如果一个机器的输出都是1，那么就已经装配完成，就用它跟汇点连边。然后n*n扫描所有机器，建图，如果一个机器的输出和另外一个机器的输入相吻合，就连边，权值是min(v[i],v[j])。
dinic求出最大流以后要输出所有弧上的变化量，所以我在建图完的时候复制了一份初始流量，再用求完最大流的流量图跟他比较，流量有减少的弧，说明是正向弧，就输出这条弧上的起点、终点、变化量。
代码：

#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<iostream>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<stack>
#include<cctype>
#include<set>
#include<cassert>
#include<algorithm>
using namespace std;
typedef long long ll;
const int INF=1e9+7;
typedef pair<int,int> pii;
int p,n,st,ed;
int v[55],pin[55][15],pout[55][15];
int pic[55][55],start[55][55];
int d[55];
void build(){
    st=0;ed=n+1;
    for(int i=1;i<=n;i++){
        bool isst=true,ised=true;
        //check st
        for(int j=1;j<=p;j++){
            if(pin[i][j]==1){//全0或者有些是2的都可以作为起点
                isst=false;break;
            }
        }
        if(isst){
            pic[st][i]=v[i];
        }
        //check ed
        for(int j=1;j<=p;j++){
            if(pout[i][j]!=1){
                ised=false;break;
            }
        }
        if(ised){
            pic[i][ed]=v[i];
        }
        //build
        for(int j=1;j<=n;j++)if(i!=j){
            bool isconnect=true;
            for(int k=1;k<=p;k++){
                if(pin[j][k]!=2&&pout[i][k]!=pin[j][k]){
                    isconnect=false;break;
                }
            }
            if(isconnect){
                pic[i][j]=min(v[i],v[j]);
            }
        }
    }
    //copy to start[55][55]
    for(int i=0;i<=ed;i++){
    //复制一份初始流量，等最大流结束就可以统计不同弧上流量的变化
        for(int j=0;j<=ed;j++){
            start[i][j]=pic[i][j];
        }
    }
}
bool BFS(){
    memset(d,-1,sizeof d);
    queue<int> Q;
    d[st]=0;Q.push(st);
    while(!Q.empty()){
        int s=Q.front();Q.pop();
        for(int i=0;i<=ed;i++){
            if(pic[s][i]>0&&d[i]<0){
                d[i]=d[s]+1;Q.push(i);
            }
        }
    }
    return d[ed]>0;
}
int DFS(int s,int t,int flow){
    if(s==t||flow==0)return flow;
    int ans=0;
    for(int i=0;i<=ed;i++){
        if(d[i]==d[s]+1&&pic[s][i]>0){
            int ff=DFS(i,t,min(flow,pic[s][i]));
            if(ff>0){
                pic[s][i]-=ff;
                pic[i][s]+=ff;
                ans+=ff;
                flow-=ff;
                if(!flow)break;
            }
        }
    }
    if(!ans)d[s]=-1;
    return ans;
}
int dinic(){
    int ans=0;
    while(BFS()){
        ans+=DFS(st,ed,INF);
    }
    return ans;
}
int main(){
//    freopen("D://input.txt","r",stdin);
    scanf("%d%d",&p,&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&v[i]);
        for(int j=1;j<=p;j++){
            scanf("%d",&pin[i][j]);
        }
        for(int j=1;j<=p;j++){
            scanf("%d",&pout[i][j]);
        }
    }
    build();
    printf("%d ",dinic());
    int cnt=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)if(i!=j){
            if(pic[i][j]<start[i][j]){
    //流量减少的弧是正向弧，所以统计比原来的弧的流量少的弧的数量
                cnt++;
            }
        }
    }
    printf("%d\n",cnt);
    if(cnt){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++)if(i!=j){
                if(pic[i][j]<start[i][j]){
                    printf("%d %d %d\n",i,j,start[i][j]-pic[i][j]);
                    //减少的量就是变化量
                }
            }
        }
    }
    return 0;
}