POJ 3436 ACM Computer Factory (網絡流）

題目

題意：
生產線是全自動化的，所以需要機器來組成生產線，給定有多少中種機器，標準ACM用電腦有多少部份，每種機器將什麼樣的ACM電腦半成品處理成什麼樣的電腦半成品(對於輸入的電腦半成品，每部分有0，1，2三種狀態：代表着 0、這部分必須沒有我才能處理，1、這部分必須有我才能處理，2、這部分有沒有我都能處理。對於輸出的電腦半成品有0，1兩種狀態：代表着0，處理完後的電腦半成品裏沒有這部分，1、處理完的電腦半成品有這部分)，每一個機器每小時可以處理Q個半成品（輸入數據中的Qi）。
輸入規格有三種情況：0,1,2
0：該部分不能存在
1：該部分必須保留
2：該部分可有可無
輸出規格有2種情況：0,1
0：該部分不存在
1：該部分存在

求組裝好的成產線的最大工作效率（每小時最多生成多少成品，成品的定義就是所有部分的狀態都是“1”），和每兩個機器之間的流量。

思路：
根據對機器的要求，來進行拆點，從而新建一個圖。添加兩個超級源s,超級匯t如果某個節點(i)的輸入部分不含1，則添加一條s->i路徑，容量爲Qi;如果某個節點(j)輸出全爲1，則添加一條j->t路徑，容量爲Qj;如果節點i的輸出與j的輸入不存在衝突(輸出與輸入對應位置的和不能爲1)，則添加一條i->j的路徑，容量爲min(Qi, Qj).

#include <iostream>
#include <cstring>
#include<cstdio>
#define LL long long
#include <queue>
const int MAXN =205;
const int MAXM=440020;
const int INF=0x3f3f3f3f;
using namespace std;
struct Edge
{
    int from;
    int to,cap,flow,next;
} edge[MAXM];
int head[MAXN],tot,gap[MAXN],d[MAXN],cur[MAXN],que[MAXN],p[MAXN];

void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}

void addedge(int u,int v,int c,int f)
{
    edge[tot]=(Edge)
    {
        u,v,c,f,head[u]
    };
    head[u] = tot++;
    edge[tot]=(Edge)
    {
        v,u,c,c,head[v]
    };
    head[v] = tot++;
}

int isap(int source,int sink,int N)
{
    memset(gap,0,sizeof(gap));
    memset(d,0,sizeof(d));
    memcpy(cur,head,sizeof(head));
    int top = 0,x = source,flow = 0;
    while(d[source] < N)
    {
        if(x == sink)
        {
            int Min = INF,inser=0;
            for(int i = 0; i < top; ++i)
            {
                if(Min > edge[p[i]].cap - edge[p[i]].flow)
                {
                    Min = edge[p[i]].cap - edge[p[i]].flow;
                    inser = i;
                }
            }
            for(int i = 0; i < top; ++i)
            {
                edge[p[i]].flow += Min;
                edge[p[i]^1].flow -= Min;
            }
            if(Min!=INF) flow += Min;
            top = inser;
            x = edge[p[top]^1].to;
            continue;
        }
        int ok = 0;
        for(int i = cur[x]; i != -1; i = edge[i].next)
        {
            int v = edge[i].to;
            if(edge[i].cap > edge[i].flow && d[v]+1 == d[x])
            {
                ok = 1;
                cur[x] = i;
                p[top++] = i;
                x = edge[i].to;
                break;
            }
        }
        if(!ok)
        {
            int Min = N;
            for(int i = head[x]; i != -1; i = edge[i].next)
            {
                if(edge[i].cap > edge[i].flow && d[edge[i].to] < Min)
                {
                    Min = d[edge[i].to];
                    cur[x] = i;
                }
            }
            if(--gap[d[x]] == 0) break;
            gap[d[x] = Min+1]++;
            if(x != source) x = edge[p[--top]^1].to;
        }
    }
    return flow;
}

int state[MAXN][2][10];
int pp,n;

int equall(int x,int y)
{
    for(int i=0;i<pp;i++)
        if(state[y][0][i]+state[x][1][i]==1)
            return 0;

    return 1;
}
int main()
{
//    freopen("in.txt","r",stdin);
    int w,flag1,flag2;
    while(~scanf("%d%d",&pp,&n))
    {
        init();
        memset(state,0,sizeof(state));

        for(int i=1;i<=n;i++)
        {
            flag1=flag2=1;
            scanf("%d",&w);
            for(int j=0;j<pp;j++){      //全是0或2 則連接源點
                scanf("%d",&state[i][0][j]);
                if(state[i][0][j]==1)
                    flag1=0;
            }

            for(int j=0;j<pp;j++){       //全是1 則連接匯點
                scanf("%d",&state[i][1][j]);
                if(state[i][1][j]==0||state[i][1][j]==2)
                    flag2=0;
            }

            addedge(i,i+n,w,0);          //拆點連自己

            if(flag1==1)
                addedge(0,i,INF,0);
            if(flag2==1)
                addedge(i+n,2*n+1,INF,0);

            for(int j=1;j<i;j++){
                if(equall(i,j))
                    addedge(n+i,j,INF,0);
                if(equall(j,i))           //注意不能是else if 兩個機器可以是雙向連接的
                    addedge(n+j,i,INF,0);
            }
        }

        printf("%d",isap(0,2*n+1,2*n+2));

        int ans[400][3];
        int ss=0;
        for(int i=n+1;i<=(n<<1);i++)      //鄰接表找最大流中的邊
            for(int j=head[i];j!=-1;j=edge[j].next)
            {
                if(edge[j].to==2*n+1||edge[j].to==i-n)
                    continue;
                if(edge[j].flow>0){
                    ans[ss][0]=edge[j].from-n;
                    ans[ss][1]=edge[j].to;
                    ans[ss++][2]=edge[j].flow;
                }
            }
        printf(" %d\n",ss);
        for(int i=0;i<ss;i++)
            printf("%d %d %d\n",ans[i][0],ans[i][1],ans[i][2]);
    }
    return 0;
}