poj 1112 染色+DP

传送门

题意：n个人分成2队，要尽量人数接近，同时一个队的人互相全部认识，问怎么分。

思路：首先任何不能在一组的建边，每个连通染色一下，看是否能构成二分图，不能则no solution，能就将存成数组，最终存成了x[1][1],x[1][2],x[2][1],x[2][2]......x[xnum][1],x[xnum][2]。之后dp，对于x[m]组中的两个，要么前面的在第一队，要么后面的，dp并存选择方案，最后输出就好。

感想：我写了染色部分，队友写了dp部分，合作愉快。（我dp太水了。。。。。）

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int fst[205],next[20005],node[20005];
int en,n,x[205][4],xnum;
bool p[205][205],vis[205],c[205],flag;
int mark[205];
int q[2005],d[205];
void init()
{
    en=0;
    xnum=0;
    flag=0;
    memset(fst,-1,sizeof(fst));
    memset(p,0,sizeof(p));
    memset(vis,0,sizeof(vis));
    memset(mark,0,sizeof(mark));
}
void add(int u,int v)
{
    next[en]=fst[u];
    fst[u]=en;
    node[en]=v;
    en++;
}
int dp[211][211];
int path[211][211];
void Mark(int i,int j)
{
    if(i==0)return;
    Mark(i-1,j-x[i][path[i][j]]);
    if(path[i][j]==1)mark[2*i]=1;
    else mark[2*i+1]=1;
}
void DP()
{
    memset(dp,-1,sizeof(dp));
    memset(path,-1,sizeof(path));
    dp[0][0]=1;
    for(int i=1;i<=xnum;i++)
    {
        for(int j=0;j<=n/2;j++)
        {
            if(dp[i-1][j]==1)
            {
                dp[i][j+x[i][1]]=1;
                path[i][j+x[i][1]]=1;
                dp[i][j+x[i][2]]=1;
                path[i][j+x[i][2]]=2;
            }
        }
    }
    int sum;
    for(int i=n/2;i>=0;i--)
    {
        if(dp[xnum][i]==1)
        {
            sum=i;
            break;
        }
    }
    Mark(xnum,sum);
}
int main()
{
    int u,v;
    while(scanf("%d",&n)!=EOF)
    {
        init();
        for(int i=1;i<=n;i++)
        {
            while(scanf("%d",&u)&&u)
            {
                p[i][u]=1;
            }
            for(int j=i-1;j>0;j--)
            {
                if(p[i][j]==0||p[j][i]==0)
                {
                    add(i,j);
                    add(j,i);
                }
            }
        }
        for(int i=1;i<=n;i++)
        {
            if(!vis[i])
            {
                xnum++;
                int cnt1=0,cnt2=0;
                int front=0,end=0;
                vis[i]=1;
                c[i]=0;
                d[i]=2*xnum;
                cnt1++;
                q[end++]=i;
                while(front<end)
                {
                    u=q[front++];
                    for(int j=fst[u];j!=-1;j=next[j])
                    {
                        v=node[j];
                        if(vis[v])
                        {
                            if(c[u]==c[v])flag=1;
                        }
                        else
                        {
                            vis[v]=1;
                            if(c[u])
                            {
                                c[v]=0;
                                d[v]=2*xnum;
                                cnt1++;
                            }
                            else
                            {
                                c[v]=1;
                                d[v]=2*xnum+1;
                                cnt2++;
                            }
                            q[end++]=v;
                        }
                    }
                }
                x[xnum][1]=cnt1;
                x[xnum][2]=cnt2;
//                cout<<cnt1<<" "<<cnt2<<endl;
            }
        }
        if(xnum==1&&(x[1][1]==0||x[1][2]==0))flag=1;
        if(flag)
        {
            cout<<"No solution"<<endl;
            continue;
        }
        DP();
        int cnt=0;
        for(int i=1;i<=n;i++)
        {
            if(mark[d[i]])cnt++;
        }
        cout<<cnt;
        for(int i=1;i<=n;i++)
        {
            if(mark[d[i]])cout<<" "<<i;
        }
        cout<<endl;
        cout<<n-cnt;
        for(int i=1;i<=n;i++)
        {
            if(mark[d[i]]==0)cout<<" "<<i;
        }
        cout<<endl;
    }
    return 0;
}