POJ 3436 ACM Computer Factory (最大流+拆点建图+输出路径)

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1e2+10;
const int inf = 0x3f3f3f3f;
int p,n,S,T,en=1,tot;
int head[maxn],h[maxn],q[maxn];
int input[100][20],output[100][20];
struct edge{
	int to,next,v,c;
}e[maxn];
inline void init()
{
	en=1;tot=0;S=0;T=2*n+1;
	memset(input,0,sizeof input);
	memset(output,0,sizeof output);
	memset(head,0,sizeof head);
	memset(e,0,sizeof e);
}
inline void insert(int u,int v,int f){
	e[++en].to=v;e[en].c=f;e[en].v=f;e[en].next=head[u];head[u]=en;
	e[++en].to=u;e[en].c=0;e[en].v=0;e[en].next=head[v];head[v]=en;
}
inline int dfs(int x,int f){
	if(x==T) return f;
	int w,used=0;
	for(int i=head[x];i;i=e[i].next){
		int v = e[i].to;
		if(h[v]==h[x]+1&&e[i].v){
			w = dfs(v,min(f-used,e[i].v));
			e[i].v-=w;
			e[i^1].v+=w;
			used+=w;
			if(used==f) return f;
		}
	}
	if(!used) h[x]=-1;
	return used;
}
inline int bfs()
{
	int f=0,t=0;
	memset(h,-1,sizeof h);
	q[t++]=S,h[S]=0;
	while(f<t){
		int now = q[f++];
		for(int i=head[now];i;i=e[i].next){
			int v=e[i].to;
			if(h[v]==-1&&e[i].v){
				h[v]=h[now]+1;
				q[t++]=v;
			}
		}
	}
	if(h[T]==-1) return 0;
	return 1;
}
inline void dinic()
{
	int m=0;
	while(bfs()) tot+=dfs(S,inf);
	for(int i=1+n;i<=n*2;i++)
	  for(int j=head[i];j;j=e[j].next)
	  if(e[j].to>0&&e[j].to<=n&&e[j].c>e[j].v) m++;
	  printf("%d %d\n",tot,m);
	  for(int i=1+n;i<=n*2;i++)
	  for(int j=head[i];j;j=e[j].next)
	  if(e[j].to>0&&e[j].to<=n&&e[j].c>e[j].v)
	  printf("%d %d %d\n",i-n,e[j].to,e[j].c-e[j].v);
}
int main()
{
	while(~scanf("%d%d",&p,&n))
	{
		init();
		for(int i=1;i<=n;++i){
			int q,flag1=0,flag2=0;scanf("%d",&q);
			insert(i,n+i,q);
			for(int j=1;j<=p;++j){
				scanf("%d",&input[i][j]);
				if(input[i][j]==1) flag1=1; 
			}
			if(!flag1) insert(S,i,inf);
			for(int j=1;j<=p;++j){
				scanf("%d",&output[i][j]);
				if(!output[i][j]) flag2=1;
			}
			if(!flag2) insert(i+n,T,inf);
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			if(i!=j){
				int flag=1;
				for(int k=1;flag&&k<=p;k++){
					if(output[i][k]==input[j][k]||input[j][k]==2) continue;
					flag=0;
				}
				if(flag) insert(i+n,j,inf);
			}
		}
		dinic();
	}
 }