poj 3436 网络流

纠结了好久的题，必须要拆点，但是有个不拆点也能A的代码。可是这组样例告诉我们必须拆点：

2 4

10 0 0 0 1

10 0 0 0 0

10 0 1 1 1

10 0 1 1 1

不拆点的代码：（AC了 但是是错的）

点与点之间的权值为min（i,j的容量）

#include<stdio.h>
#include<string.h>
#define MX 1000000
int n,p,map[200][200],pre[200],vs,vt,mapcy[200][200],transport[200][5];
struct point{
	int numb,in[15],out[15];
}machine[1000];
int min(int a,int b){
	return a<b?a:b;
}
int minflow(){
	int flow=0;
	while(1){
		for(int i=0;i<=vt;i++){
			pre[i]=-1;
		}
		int qs=0,qe=1;
		int queue[1000];
		queue[qs]=vs;
		while(qs<qe){
			int v=queue[qs++];
			for(int i=1;i<=vt;i++){
				if(map[v][i]&&pre[i]==-1){
					pre[i]=v;
					queue[qe++]=i;
				}
			}
		}
	//	printf("!!prevt=%d\n",pre[vt]);
		if(pre[vt]==-1){
			break;
		}
		int minf=MX;
		for(int i=pre[vt],j=vt;i!=-1;){
			if(map[i][j]<minf){
				minf=map[i][j];
			}
			j=i;
			i=pre[j];
		}
		for(int i=pre[vt],j=vt;i!=-1;){
			map[i][j]-=minf;
			map[j][i]+=minf;
			j=i;
			i=pre[j];
		//	printf("! i=%d j=%d pre[j]=%d minf=%d\n",i,j,pre[j],minf);
		}
		flow+=minf;
	//	printf("!\n");
	}
	return flow;
}
int main(){
	while(scanf("%d%d",&p,&n)!=EOF){
		memset(map,0,sizeof(map));
		for(int i=1;i<=n;i++){
			scanf("%d",&machine[i].numb);
			for(int j=1;j<=p;j++){
				scanf("%d",&machine[i].in[j]);
			}
			for(int j=1;j<=p;j++){
				scanf("%d",&machine[i].out[j]);
			}
		}
		vs=0;
		vt=n+1;
		for(int i=1;i<=n;i++){
			bool flag_s=true,flag_t=true;
			for(int j=1;j<=p;j++){
				if(machine[i].in[j]==1){
					flag_s=false;
				}
				if(machine[i].out[j]==0){
					flag_t=false;
				}
			}
			if(flag_s){
				map[vs][i]=machine[i].numb;
			}
			if(flag_t){
				map[i][vt]=machine[i].numb;
			}
			for(int k=1;k<=n;k++){
				if(i!=k){
					bool flag=true;
					for(int j=1;j<=p;j++){
						if(machine[i].out[j]+machine[k].in[j]==1){
							flag=false;
						}
					}
					if(flag){
						map[i][k]=min(machine[i].numb,machine[k].numb);
					}
				}
			}
		}
		memcpy(mapcy,map,sizeof(map));
		int flow=minflow(),pos=0;
		printf("%d",flow);
		for(int i=1;i<vt;i++){
			for(int j=1;j<vt;j++){
				if(mapcy[i][j]>map[i][j]){
					transport[pos][0]=i;
					transport[pos][1]=j;
					transport[pos++][2]=mapcy[i][j]-map[i][j];
				}
			}
		}
		printf(" %d\n",pos);
		for(int i=0;i<pos;i++){
			printf("%d %d %d\n",transport[i][0],transport[i][1],transport[i][2]);
		}
	}
	return 0;
}

拆点的做法：

状态全为0或2的源点相连，权值为MX。状态全为1或2的与汇点相连，权值也为MX。

点i与拆开的点i+n权值为其点的容量。点与其能连的点的权值为MX。

#include<stdio.h>
#include<string.h>
#define MX 1000000
int n,p,map[400][400],pre[400],vs,vt,mapcy[400][400],transport[400][5];
struct point{
	int numb,in[15],out[15];
}machine[1000];
int min(int a,int b){
	return a<b?a:b;
}
int minflow(){
	int flow=0;
	while(1){
		for(int i=0;i<=vt;i++){
			pre[i]=-1;
		}
		int qs=0,qe=1;
		int queue[1000];
		queue[qs]=vs;
		while(qs<qe){
			int v=queue[qs++];
			for(int i=1;i<=vt;i++){
				if(map[v][i]&&pre[i]==-1){
					pre[i]=v;
					queue[qe++]=i;
				}
			}
		}
	//	printf("!!prevt=%d\n",pre[vt]);
		if(pre[vt]==-1){
			break;
		}
		int minf=MX;
		for(int i=pre[vt],j=vt;i!=-1;){
			if(map[i][j]<minf){
				minf=map[i][j];
			}
			j=i;
			i=pre[j];
		}
		for(int i=pre[vt],j=vt;i!=-1;){
			map[i][j]-=minf;
			map[j][i]+=minf;
			j=i;
			i=pre[j];
		//	printf("! i=%d j=%d pre[j]=%d minf=%d\n",i,j,pre[j],minf);
		}
		flow+=minf;
	//	printf("!\n");
	}
	return flow;
}
int main(){
	while(scanf("%d%d",&p,&n)!=EOF){
		memset(map,0,sizeof(map));
		for(int i=1;i<=n;i++){
			scanf("%d",&machine[i].numb);
			for(int j=1;j<=p;j++){
				scanf("%d",&machine[i].in[j]);
			}
			for(int j=1;j<=p;j++){
				scanf("%d",&machine[i].out[j]);
			}
		}
		vs=0;
		vt=2*n+1;
		for(int i=1;i<=n;i++){
			bool flag_s=true,flag_t=true;
			for(int j=1;j<=p;j++){
				if(machine[i].in[j]==1){
					flag_s=false;
				}
				if(machine[i].out[j]==0){
					flag_t=false;
				}
			}
			if(flag_s){
				map[vs][i]=machine[i].numb;
			}
			if(flag_t){
				map[i+n][vt]=machine[i].numb;
			}
			for(int k=1;k<=n;k++){
				if(i!=k){
					bool flag=true;
					for(int j=1;j<=p;j++){
						if(machine[i].out[j]+machine[k].in[j]==1){
							flag=false;
						}
					}
					if(flag){
						map[i+n][k]=MX;
					}
				}
			}
		}
		for(int i=1;i<=n;i++){
			map[i][i+n]=machine[i].numb;
		}
		memcpy(mapcy,map,sizeof(map));
		int flow=minflow(),pos=0;
		printf("%d",flow);
		for(int i=1;i<vt;i++){
			for(int j=1;j<vt;j++){
				if(mapcy[i][j]>map[i][j]&&map[i][j]!=MX&&i!=j+n&&j!=i+n){
					if(i>n){
						transport[pos][0]=i-n;
					}
					else{
						transport[pos][0]=i;
					}
					if(j>n){
						transport[pos][1]=j-n;
					}
					else{
						transport[pos][1]=j;
					}
					transport[pos++][2]=mapcy[i][j]-map[i][j];
				}
			}
		}
		printf(" %d\n",pos);
		for(int i=0;i<pos;i++){
			printf("%d %d %d\n",transport[i][0],transport[i][1],transport[i][2]);
		}
	}
	return 0;
}