poj 1112

昨天晚上看的题。

说实话，我一眼就看出了是二分图，再一眼就看出了是二分图+dp（01背包）。但悲剧的是我一眼看出的算法是正确的，但我总以为它是错误的，浪费了很长时间像其他算法（TAT）。

今天终于把代码打了出来，刚开始01背包的优化后来发现是错误的，忘记删了，导致WA一次。吃晚饭时突然发现了，便AC。

2013-6-29

代码：

#include<cstdio>

#include<cstring>

using namespace std;

 

int n,col[101],w[202],m=0,dp[101][101],pre[101][101];

bool map[101][101]={0},vis[202]={0},f=0;

void dfs(int x,int cur){

         vis[x]=1,col[x]=2*m+cur;

         w[2*m+cur]++;

         for(int i=0;i<n;i++){

                   if(map[x][i] && vis[i] && col[x]==col[i]){

                            f=1;

                            return;

                   }

                   if(map[x][i] && !vis[i]){

                            dfs(i,1^cur);

                            if(f)return;

                   }

         }

         return;

}

int max(int x,int y){

         return (x>y)?x:y;

}

int min(int x,int y){

         return (x<y)?x:y;

}

int main(){

         scanf("%d",&n);

         for(int i=0;i<n;i++){

                   int x;

                   while(scanf("%d",&x),x!=0)

                            map[i][--x]=1;

         }

         for(int i=0;i<n;i++)

                   for(int j=0;j<i;j++)

                            if(map[i][j]*map[j][i]==0 && i!=j)map[i][j]=map[j][i]=1;else map[i][j]=map[j][i]=0;

         for(int i=0;i<n;i++)if(!vis[i]){

                   dfs(i,0);

                   m++;

         }

         if(f){

                   printf("No solution\n");

                   return 0;

         }

         m*=2;

         memset(dp[0],-1,sizeof(dp[0]));

         memset(vis,0,sizeof(vis));

         dp[0][0]=0;

         for(int i=1;i<=m;i+=2)

                   for(int j=0;j<=n;j++){       //这里我一开始打成了for(int j=min(w[i-1],w[i]);j<=n;j++)

                            int a=(i+1)/2;

                            dp[a][j]=-1;pre[a][j]=-1;

                            if(j>=w[i-1] && dp[a-1][j-w[i-1]]+w[i-1]>max(dp[a][j],w[i-1]-1))dp[a][j]=max(dp[a][j],dp[a-1][j-w[i-1]]+w[i-1]),pre[a][j]=2;

                            if(j>=w[i] && dp[a-1][j-w[i]]+w[i]>max(dp[a][j],w[i]-1))dp[a][j]=max(dp[a][j],dp[a-1][j-w[i]]+w[i]),pre[a][j]=1;

                   }

         int j;

         for(j=n/2;j>=0 && dp[m/2][j]==-1;j--);

         if(j<0){

                   printf("No solution\n");

                   return 0;

         }

         int j1=j;

         for(int i=m/2;i>0;i--)

                   vis[i*2-pre[i][j]]=1,j-=w[i*2-pre[i][j]];

         printf("%d",j1);

         for(int i=0;i<n;i++)

                   if(vis[col[i]])printf(" %d",i+1);

         printf("\n");

         printf("%d",n-j1);

         for(int i=0;i<n;i++)

                   if(!vis[col[i]])printf(" %d",i+1);

         printf("\n");

         return 0;

}