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;
}