首页 > 代码库 > 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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。