UVA 1627 Team them up!(dp+二分图染色)

思路:
1.将所有不能在一组的两个人连边,二分图染色,若发现染色不成功,则无解。
2.若染色成功,同一连通分量里的黑点与白点无法分为一组,不同连通分量中的点可与其他连通分量中的任一种组合,dp。
3.设dp[i][j]表示前i个连通分量,是否有差值为j的组合(j可正可负,代表A大于B或B大于A)。设同一连通分量i中的两种点数差值为diff[i],则若dp[i-1][j]==1,则dp[i][j-diff[i]]=1,dp[i][j+diff[i]]=1(即分到AB或BA)。最后判断是否存在最小的abs(j)使得dp[num][j]==1。
4.输出方案,从后向前判断设当前差值为ans,若dp[i-1][ans-diff[i]]==1成立代表将第i个连通分量两种点分为AB两组,同时ans-=diff[i];若dp[i-1][ans+diff[i]]==1成立代表将第i个连通分量中两种点分为BA两组,同时ans+=diff[i],记录即可。
 

#include<bits/stdc++.h> 

using namespace std;

const int maxn=400+50;

#define INF 0x3f3f3f3f

vector<int> G[maxn];
int g[maxn][maxn],n,num;
int color[maxn],vis[maxn];
vector<int> group[maxn][2];
int d[maxn][maxn],diff[maxn];

void init()
{
    num=0;
    memset(g,0,sizeof(g));
    memset(d,0,sizeof(d));
    memset(vis,0,sizeof(vis));
    memset(color,-1,sizeof(color));
    for(int i=0; i<=n; i++){
        G[i].clear();
        group[i][0].clear();
        group[i][1].clear();
    }
}

int dfs(int u,int num,int id)
{
    color[u]=id;
    group[num][id].push_back(u);
    for(int i=0; i<G[u].size(); i++){
        int nt=G[u][i];
        if(color[u]==color[nt]) return 0;
        if(!vis[nt]){
            vis[nt]=1;
            if(!dfs(nt,num,id^1)) return 0;
        }
    }
    return 1;
}

void print(int ans)
{
    int flag;
    vector<int> term1,term2;
    for(int i=num; i>=1; i--){
        if(d[i-1][ans-diff[i]+2*n]){
            flag=0;
            ans-=diff[i];
        }
        else if(d[i-1][ans+diff[i]+2*n]){
            flag=1;
            ans+=diff[i];
        }
        for(int j=0; j<group[i][flag].size(); j++){
            term1.push_back(group[i][flag][j]);
        }
        for(int j=0; j<group[i][flag^1].size(); j++){
            term2.push_back(group[i][flag^1][j]);
        }
    }
    printf("%d",term1.size());
    for(int i=0; i<term1.size(); i++){
        printf(" %d",term1[i]);
    }
    printf("\n");
    printf("%d",term2.size());
    for(int i=0; i<term2.size(); i++){
        printf(" %d",term2[i]);
    }
    printf("\n");
}

int main()
{
    int t,cas=0;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        init();
        for(int i=1; i<=n; i++){
            int x;
            while(scanf("%d",&x)!=EOF&&x) g[i][x]=1;
        }
        for(int i=1; i<=n; i++){
            for(int j=1; j<=n; j++){
                if(i==j) continue;
                if(!g[i][j]||!g[j][i]){
                    G[i].push_back(j);
                }
            }
        }
        int flag=0;
        for(int i=1; i<=n; i++){
            if(!vis[i]){
                vis[i]=1;
                if(!dfs(i,++num,0)){
                    flag=1;
                    break;
                }
            }
        }
        if(++cas!=1) printf("\n");
        if(flag) printf("No solution\n");
        else{
            d[0][2*n]=1;
            for(int i=1; i<=num; i++){
                diff[i]=group[i][0].size()-group[i][1].size();
                for(int j=n; j>=-n; j--){

                    if(d[i-1][j+2*n]){
                        d[i][j-diff[i]+2*n]=1;
                        d[i][j+diff[i]+2*n]=1;
                    }
                }
            }
            int minx=INF,Diff;	
            for(int i=-n; i<=n; i++){
                if(d[num][i+2*n]&&abs(i)<minx){
                   minx=abs(i);
                   Diff=i;
                }
            }
            print(Diff);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40679299/article/details/83994576