poj 3436 ACM Computer Factory (最大流)

题目真心没读懂，参考这位仁兄

通过拆点建图，然后跑一边最大流，枚举每条边找出有流量减少的边，那么这条边肯定是最大流上的边输出即可。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define B(x) (1<<(x))
using namespace std;
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const int MOD=1000007;
const int maxn=5005;
struct EDGE{
    int v,next,c,f,id;
}E[maxn];
int head[maxn],tol;
int gap[maxn],dep[maxn],pre[maxn],cur[maxn];

void Init(){
    memset(head,-1,sizeof head);
    tol=0;
}

void add_edge(int u,int v,int f,int id,int rf=0){
    E[tol].v=v;
    E[tol].c=f;
    E[tol].f=0;
    E[tol].id=id;
    E[tol].next=head[u];
    head[u]=tol++;

    E[tol].v=u;
    E[tol].c=rf;
    E[tol].f=0;
    E[tol].id=id;
    E[tol].next=head[v];
    head[v]=tol++;
}

int isap(int start,int end,int N){
    memset(gap,0,sizeof gap);
    memset(dep,0,sizeof dep);
    memcpy(cur,head,sizeof head);
    int u=start;
    pre[u]=-1;
    gap[0]=N;
    int ans=0;
    while(dep[start]<N){
        if(u==end){
            int Min=oo;
            for(int i=pre[u];i!=-1;i=pre[E[i^1].v])
                if(Min>E[i].c-E[i].f)
                    Min=E[i].c-E[i].f;
            for(int i=pre[u];i!=-1;i=pre[E[i^1].v]){
                E[i].f+=Min;
                E[i^1].f-=Min;
            }
            u=start;
            ans+=Min;
            continue;
        }
        bool flag=false;
        int v;
        for(int i=cur[u];i!=-1;i=E[i].next){
            v=E[i].v;
            if(E[i].c-E[i].f&&dep[v]+1==dep[u]){
                flag=true;
                cur[u]=pre[v]=i;
                break;
            }
        }
        if(flag){
            u=v;
            continue;
        }
        int Min=N;
        for(int i=head[u];i!=-1;i=E[i].next)
            if(E[i].c-E[i].f&&dep[E[i].v]<Min){
                Min=dep[E[i].v];
                cur[u]=i;
            }
        gap[dep[u]]--;
        if(!gap[dep[u]])return ans;
        dep[u]=Min+1;
        gap[dep[u]]++;
        if(u!=start)u=E[pre[u]^1].v;
    }
    return ans;
}

int part[55][22],w[55];
struct PATH{
    int u,v,w;
}a[maxn];

int main(){
    //freopen("G:\\read.txt","r",stdin);
    int N,P,S,T;
    while(scanf("%d %d",&P,&N)!=EOF){
        Init();
        for(int i=1;i<=N;i++){
            scanf("%d",&w[i]);
            for(int j=1;j<=2*P;j++){
                scanf("%d",&part[i][j]);
            }
        }
        S=2*N+1;
        T=S+1;
        for(int i=1;i<=N;i++){
            int f=1;
            for(int j=1;j<=P;j++)if(part[i][j]==1)f=0;
            if(f)add_edge(S,i,oo,i);
            f=1;
            for(int j=P+1;j<=2*P;j++)if(part[i][j]!=1)f=0;
            if(f)add_edge(i+N,T,oo,oo);
        }
        for(int i=1;i<=N;i++){
            add_edge(i,i+N,w[i],i);
            for(int j=1;j<=N;j++){
                if(i==j)continue;
                int f=1;
                for(int k=1;k<=P;k++)
                    if(part[i][k+P]+part[j][k]==1)
                        f=0;
                if(f)add_edge(i+N,j,oo,j);
            }
        }
        int ans=isap(S,T,T);
        int cnt=0;
        for(int i=N+1;i<=2*N;i++){
            for(int j=head[i];j!=-1;j=E[j].next){
                if(i==E[j].id||E[j].id==oo)continue;
                if(E[j].f>0){
                    a[cnt].u=i-N;
                    a[cnt].v=E[j].id;
                    a[cnt].w=E[j].f;
                    cnt++;
                }
            }
        }
        printf("%d %d\n",ans,cnt);
        for(int i=0;i<cnt;i++){
            printf("%d %d %d\n",a[i].u,a[i].v,a[i].w);
        }
    }
    return 0;
}
/*
3 4
15  0 0 0  0 1 0
10  0 0 0  0 1 1
30  0 1 2  1 1 1
3   0 2 1  1 1 1
3 5
5   0 0 0  0 1 0
100 0 1 0  1 0 1
3   0 1 0  1 1 0
1   1 0 1  1 1 0
300 1 1 2  1 1 1
2 2
100  0 0  1 0
200  0 1  1 1

3 11
10  2 2 2 0 0 1
20  2 2 2 0 1 1
30  2 2 2 0 0 1
40  2 2 2 0 1 1
50  2 2 2 0 0 1
1000 2 2 1 1 0 0
50  1 2 2 1 1 1
40  1 2 0 1 1 1
30  1 0 2 1 1 1
20  1 0 0 1 1 1
10  1 2 2 1 1 1
*/