poj 1112 염색약 + DP

전송문
제목: n 개인은 2팀으로 나뉘어 최대한 인원수에 가까워야 하며, 동시에 한 팀의 사람들은 서로를 모두 알고 어떻게 나눌지 물어야 한다.
사고방식: 우선 한 그룹의 가장자리를 만들 수 없고 하나하나가 연결되어 염색을 한다. 이분도를 구성할 수 있는지 없는지를 보고 nosolution을 할 수 없으면 수조로 저장할 수 있다. 최종적으로 x[1], x[1], x[2], x[2], x[2], x[2][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;
}