POJ 1112 Team Them Up!【二分图染色+DP】

题意： 有 n 个人，他们之间的关系有四种，给出一些关系 a，b 表示b 知道 a，现在想把这些人分成两组，每个组里面所有人都相互知道，如果可以分成这两组，

         找出两组人数相差最少的情况。

分析：如果 a 和 b 不是相互知道，就在a，b之间连一条双向边，表示a 和b 绝不能分在一个组里

         建好图之后，进行染色，判断是否是二分图，如果不是二分图，肯定不存在符合条件的情况

         染色的同时，记录每个连通块中每个部分的个数，并记录路径

         用 01 背包标记所有存在的状态，找到差值最小的情况

#include<stdio.h>
#include<string.h>
#define clr(x)memset(x,0,sizeof(x))
#define maxn 110
int g[maxn][maxn];
int r[maxn][maxn];
int dp[maxn][maxn];
int c[maxn];
int num[maxn][2];
int f[maxn];
int flag;
int sn,d,n;
void dfs(int x)
{
    int i;
    for(i=1;i<=n;i++)
        if(r[x][i])
        {
            if(c[x]==c[i])
            {
                flag=1;
                return;
            }
            else if(c[i]!=-1)
                continue;
            f[i]=sn;
            c[i]=c[x]^1;
            num[sn][c[i]]++;
            dfs(i);
            f[i]=sn;
            if(flag==1)
                return;
        }
}
int main()
{
    int i,j,p;
    while(scanf("%d",&n)!=EOF)
    {
        clr(g);
        clr(r);
        clr(f);
        clr(num);
        clr(f);
        clr(dp);
        memset(c,-1,sizeof(c));
        for(i=1;i<=n;i++)
        {
            while(scanf("%d",&p),p)
                g[i][p]=1;    
        }
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
            {
                if(g[i][j]&&g[j][i])
                    continue;
                if(i==j)
                    continue;
                r[i][j]=r[j][i]=1;
            }
        sn=1;
        flag=0;
        for(i=1;i<=n;i++)
            if(c[i]==-1)
            {
                c[i]=0;
                num[sn][0]++;
                f[i]=sn;
                dfs(i);
                if(flag==1)
                {
                    printf("No solution\n");
                    break;
                }
                sn++;
            }
        if(flag)
            return 0;
        dp[0][0]=1;
        for(i=1;i<sn;i++)
            for(j=0;j<=n/2;j++)
            {
                if(dp[i-1][j])
                {
                    if(j+num[i][0]<=n/2)
                        dp[i][j+num[i][0]]=1;
                    if(j+num[i][1]<=n/2)
                        dp[i][j+num[i][1]]=1;
                }
            }
        int tmp,cn;
        for(i=n/2;i>=0;i--)
        {
            if(dp[sn-1][i]==1)
            {
                tmp=i;
                cn=i;
                break;
            }
        }
        int vis[maxn];
        clr(vis);
        for(i=sn-1;i>=1;i--)
        {
            int tt;
            if(tmp-num[i][0]>=0&&dp[i-1][tmp-num[i][0]])
            {
                tt=0;
                tmp=tmp-num[i][0];
            }
            else if(tmp-num[i][1]>=0&&dp[i-1][tmp-num[i][1]])
            {
                tmp=tmp-num[i][1];
                tt=1;
            }
            for(j=1;j<=n;j++)
                if(f[j]==i&&c[j]==tt)
                    vis[j]=1;
        }
        printf("%d ",cn);
        for(i=1;i<=n;i++)
            if(vis[i])
                printf("%d ",i);
        putchar('\n');
        printf("%d ",n-cn);
        for(i=1;i<=n;i++)
            if(!vis[i])
                printf("%d ",i);
        putchar('\n');
    }
    return 0;
}