首页 > 代码库 > poj 1112 Team Them Up! 二分图染色+dp

poj 1112 Team Them Up! 二分图染色+dp

题意:

给n个人和一些认识关系,要将这n个人分成两队,每队的人之间都互相认识,求一种方案使两队的人数差最小。

分析:

对原图求逆得到新图g,g中如果有边(a,b),那么a,b不能在一个队,对新图进行二分图染色就能求得一种方案了。但题目要使人数差最小,所以还要dp一下。dp[i][j]表示前i个连通分量获得j个人的队伍是否可行,这其实是个背包问题,每个物品有多种重量,问是否能获得一个特定的重量和。

代码:

//poj 1112
//sep9
#include <iostream>
using namespace std;
const int maxN=128;
bool g[maxN][maxN],dp[maxN][maxN/2];
int n,cnt;
int color[maxN],p[maxN][2][maxN],num[maxN][2],path[maxN][maxN];

int dfs(int u)
{
	int col=color[u];
	int index=num[cnt][col];	
	++num[cnt][col];
	p[cnt][col][index]=u;
	for(int i=1;i<=n;++i)
		if(g[u][i]==true){
			if(color[i]==-1){
				color[i]=col^1;
				if(!dfs(i))
					return false;
			}
			else if(color[i]==color[u])
				return false;		
		}
	return true;
}

int main()
{
	int i,j,x,tmp,res;
	scanf("%d",&n);
	memset(g,false,sizeof(g));	
	for(i=1;i<=n;++i)
		while(scanf("%d",&x)==1&&x)
			g[i][x]=true;		
	for(i=1;i<=n;++i)
		for(j=i+1;j<=n;++j){
			if(g[i][j]&&g[j][i])
				g[i][j]=g[j][i]=false;
			else
				g[i][j]=g[j][i]=true;
		}
		
	cnt=0;
	memset(color,-1,sizeof(color));
	memset(num,0,sizeof(num));
	for(i=1;i<=n;++i){
		if(color[i]==-1){
			color[i]=0;
			++cnt;
			if(!dfs(i)){
				printf("No solution");
				return 0;
			}
		}
	}	
	memset(dp,false,sizeof(dp));
	dp[0][0]=true;
	for(i=1;i<=cnt;++i)
		for(j=0;j<=n/2;++j){
			tmp=j-num[i][0];
			if(tmp>=0&&dp[i-1][tmp]){
				dp[i][j]=true;
				path[i][j]=0;
			}
			tmp=j-num[i][1];
			if(tmp>=0&&dp[i-1][tmp]){
				dp[i][j]=true;
				path[i][j]=1;
			}	
		}
	
	int ans,col;
	for(i=n/2;i>0;--i)
		if(dp[cnt][i])
			break;
	ans=i;
	tmp=i;	
	printf("%d",ans);		
	for(i=cnt;i>=1;--i){
		if(path[i][ans]==0){
			col=0;
			ans-=num[i][0];
		}else{
			col=1;
			ans-=num[i][1];
		}
		for(j=0;j<num[i][col];++j)
			printf(" %d",p[i][col][j]);
	}
	ans=tmp;
	printf("\n%d",n-ans);		
	for(i=cnt;i>=1;--i){
		if(path[i][ans]==0){
			col=0;
			ans-=num[i][0];
		}else{
			col=1;
			ans-=num[i][1];
		}
		for(j=0;j<num[i][col^1];++j)
			printf(" %d",p[i][col^1][j]);
	}		
	return 0;	
} 


poj 1112 Team Them Up! 二分图染色+dp