[POJ3436]ACM Computer Factory 做题笔记

题目链接：http://poj.org/problem?id=3436
拆点网络流典型题，每个机器拆成两个点。
有点值得注意的地方：如果是int a[N][N] 这样的数组，a[i][j]是数组中的数据，而a[i]则还是a[i][0]的指针，与char数组不同的一点，int数组不能随便用&取地址，所以下文标记处不能用&in1[i]。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N=200,M=10000,inf=0x3fffffff;

int head[N],cap[N],d[N],e[M<<1],b[M<<1],ver[M<<1],next[M<<1];
int s,t,tot=1,n,k,maxflow=0,ans=0;
int sta[20],end[20];
int in1[N][20],out1[N][20];
int disp[M<<1][3];
bool z[M<<1];

void add (int u,int v,int w,bool flag=0) {
    ver[++tot]=v;e[tot]=b[tot]=w;z[tot]=flag;next[tot]=head[u];head[u]=tot;
    ver[++tot]=u;e[tot]=b[tot]=0;z[tot]=flag;next[tot]=head[v];head[v]=tot;
}

bool cmp (int *a,int *b) {
    for (int i=0;i<k;i++) {
        if (a[i]==2||b[i]==2) continue;
        if (a[i]!=b[i]) return 0;
    }
    return 1;
}

bool bfs () {
    queue<int> q;
    memset(d,0,sizeof(d));
    q.push(s); d[s]=1;
    while (!q.empty()) {
        int x=q.front(); q.pop();
        for (int i=head[x];i;i=next[i]) 
            if (e[i]&&!d[ver[i]]) {
                q.push(ver[i]);
                d[ver[i]]=d[x]+1;
                if (ver[i]==t) return 1;
            }
    }
    return 0;
}

int dinic (int x,int f) {
    int rest=f;
    if (x==t) return f;
    for (int i=head[x];i&&rest;i=next[i]) 
        if (e[i]&&d[ver[i]]==d[x]+1) {
            int now=dinic(ver[i],min(e[i],rest));
            if (!now) d[ver[i]]=0;//
            e[i]-=now;
            e[i^1]+=now;
            rest-=now;
        }
    return f-rest;
}

int main () {
    int tmp;
    scanf("%d%d",&k,&n);
    s=0,t=2*n+1;
    for (int i=0;i<k;i++) end[i]=1;
    for (int i=1;i<=n;i++) {
        scanf("%d",&cap[i]);
        for (int j=0;j<k;j++) 
            scanf("%d",&in1[i][j]);
        for (int j=0;j<k;j++)
            scanf("%d",&out1[i][j]);
    }
    for (int i=1;i<=n;i++) {
        add(i,i+n,cap[i],1);
        if (cmp(sta,in1[i]))//这是标记=。=
            add(s,i,inf,1);
    }
    for (int i=1;i<=n;i++) {
        if (cmp(end,out1[i])) {
            add(i+n,t,inf,1);continue;
        }
        for (int j=1;j<=n;j++) 
            if (cmp(out1[i],in1[j])&&i!=j) {
                add(i+n,j,inf);
            }
    }
    while (bfs())
        while (tmp=dinic(s,inf)) maxflow+=tmp;
    printf("%d ",maxflow);
    for (int i=2;i<=tot;i+=2) {
        if (!z[i]&&e[i]!=b[i]) {
            disp[++ans][1]=ver[i];
            disp[ans][0]=ver[i^1]-n;
            disp[ans][2]=b[i]-e[i];
        }
    }
    printf("%d\n",ans);
    for (int i=1;i<=ans;i++) {
        printf("%d %d %d\n",disp[i][0],disp[i][1],disp[i][2]);
    }
}