[POJ3436]ACM Computer Factory 做题笔记

题目链接：http://poj.org/problem?id=3436

拆点网络流典型题，每个机器拆成两个点。

有点值得注意的地方：如果是int a

这样的数组，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
,cap
,d
,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
[20],out1
[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]);
}
}