Description

Input

Output

Sample Input

5
2 3 5 0
1 4 5 3 0
1 2 5 0
1 2 3 0
4 3 2 1 0

Sample Output

3 1 3 5
2 2 4

題意：

思路：首先得判斷出誰和誰一定不能在一組，很自然地想到二分圖染色。只有兩個人相互認識才能在同一個隊伍。所以針對不能在同一個隊伍地人連邊，然後二分圖染色。

染色的過程中得統計不同顏色有多少人，哪些人。分成不同的連通塊。

然後問題就轉化成了，比如有m個連通塊，x[i]和y[i]代表不同的顏色。對於每個連通塊，需要在x和y中選一個，統計總人數，使得最後兩組人數差最小。

我覺得我寫得好像很複雜，但因為資料量很小所以也能過。

dp[i][j]表示一組i個人，一組j個人的狀態能否達到。

ans[i][j][k]表示該狀態可以達到的情況下，第k個連通塊選了x還是y。

最後再算一下最小值，更新結果。

寫得太浪了。。。

#include <stdio.h>
#include <string.h>
#include <vector>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAXN 110
#define INF 1000000007

vector <int> G[MAXN];
int V;
int color[MAXN];
vector <int> x[MAXN];
vector <int> y[MAXN];
int d[MAXN][MAXN];
int dp[MAXN][MAXN];
int ans[MAXN][MAXN][MAXN];
int num=0;
bool dfs(int v,int c){
    color[v]=c;
    if(c>0)
        x[num].push_back(v);
    else
        y[num].push_back(v);
    for(int i=0;i<G[v].size();i++){
        if(color[G[v][i]]==c)
            return false;
        if(color[G[v][i]]==0&&!dfs(G[v][i],-c)){
            return false;
        }
    }
    return true;
}

int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int t;
        while(scanf("%d",&t)&&t){
            d[i][t]=1;
        }
    }
    for(int i=1;i<n;i++){
        for(int j=i+1;j<=n;j++){
            if(!d[i][j]||!d[j][i])
            {
                G[i].push_back(j);
                G[j].push_back(i);
            }
        }
    }
    for(int i=1;i<=n;i++){
        if(!color[i]){
            if(dfs(i,1)){
                num++;
            }
            else{
                printf("No solution\n");
                return 0;
            }
        }
    }
    memset(dp,0,sizeof(dp));
    vector <int>r1;
    vector <int> r2;
    int m=n/2;
    dp[0][0]=1;
    memset(ans,-1,sizeof(ans));
    for(int k=0;k<num;k++){
        for(int i=0;i<=n;i++){
            for(int j=0;j<=n;j++){
                if(dp[i][j]&&ans[i][j][k]==-1){
                    int a=x[k].size(),b=y[k].size();
                    if(dp[i+a][j+b]==0){
                        dp[i+a][j+b]=1;
                        ans[i+a][j+b][k]=0;
                        //printf("<<<%d %d %d >>>\n",i+a,j+b,ans[i+a][j+b][k]);
                        for(int t=0;t<k;t++)
                            ans[i+a][j+b][t]=ans[i][j][t];
                    }
                    if(dp[i+b][j+a]==0){
                        dp[i+b][j+a]=1;
                        ans[i+b][j+a][k]=1;
                        //printf("<<<%d %d %d >>>\n",i+b,j+a,ans[i+b][j+a][k]);
                        for(int t=0;t<k;t++)
                            ans[i+b][j+a][t]=ans[i][j][t];
                    }
                }
            }
        }
    }
    int min_f=INF;
    for(int i=1;i<=n/2;i++){
        int j=n-i;
        if(dp[i][j]){
            if((j-i)<min_f){
                r1.clear();
                r2.clear();
                for(int k=0;k<num;k++){
                    if(ans[i][j][k]==0){
                        for(int s=0;s<x[k].size();s++){
                            r1.push_back(x[k][s]);
                        }
                        for(int s=0;s<y[k].size();s++){
                            r2.push_back(y[k][s]);
                        }
                    }
                    else{
                        for(int s=0;s<y[k].size();s++){
                            r1.push_back(y[k][s]);
                        }
                        for(int s=0;s<x[k].size();s++){
                            r2.push_back(x[k][s]);
                        }
                    }
                }
            }
        }
    }
    printf("%d ",r1.size());
    for(int i=0;i<r1.size();i++)
        printf(" %d",r1[i]);
    printf("\n");
    printf("%d ",r2.size());
    for(int i=0;i<r2.size();i++)
        printf(" %d",r2[i]);
    printf("\n");
}