JZOJ 3782. 【NOIP2014模拟8.17】组队

题目

你的任务是将一群人分到两个队伍中，使得：
1、每个人都属于一个队伍。
2、每个队伍至少有一个人。
3、每个队伍的任意一个人都认识其他人。
4、两支队伍的人数尽可能接近。
这个任务可能有多组解，你可以输出任意一种。
注意：认识是单向的且没有传递性。
N≤100

题解

当时做的时候，直接暴力加点，维护两个队伍，最后微调两队中的人。
但这样是不行的。因为完全可能有可能x跟两个队伍的人都有不互相认识的，但若将两队人员进行微调，x是可以归入其中一队的。
上述算法没有考虑到有连续2增广的情况。
那么这时思路就要偏向01染色，二分图匹配，如果可以DP的话，可设$f[i][j]$为第一边i人第二边j人的状态。
如果将不互相认识的人连边，进行01染色，如果一条边的两个点同色，则无解。
有一种可行的方案，是设$f[i][j]$表示做到第i个连通块，当前白点个数为j的0/1/2状态。0表示不可行，1表示该连通块所有点的颜色不改变，2表示表示该连通块所有点的颜色取反。
$f[i][j]$还可以维护整个连通块的状态。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 105
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
struct note{
    int to,next;
};note edge[N*N];
struct ltk{
    int x,g0,g1;
};ltk c[N];
struct nxt{
    int x,y,z;
};nxt g[N][N];
int tot,head[N];
int i,j,k,l,n,m,x,y,z,x2,y2,cnt;
int map[N][N],a0[N],a1[N];
int f[N][N];
int col[N]; 
bool bz[N],ia;
void lb(int x,int y){
    edge[++tot].to=y;
    edge[tot].next=head[x];
    head[x]=tot;
}
void dfs(int x){
    if(col[x]==0)c[cnt].g0++;else c[cnt].g1++;
    int i;
    for(i=head[x];i;i=edge[i].next){
        if(ia)return;
        if(col[edge[i].to]==-1){
            col[edge[i].to]=col[x]^1;
            dfs(edge[i].to);
        }else
        if(!(col[edge[i].to]^col[x])){
            ia=1;
            return;
        }
    }
}
void dfs1(int x,int z){
    bz[x]=1;
    if(z==1){
        if(col[x])a0[++a0[0]]=x;
             else a1[++a1[0]]=x;
    }else{
        if(col[x])a1[++a1[0]]=x;
             else a0[++a0[0]]=x;
    }
    int i;
    for(i=head[x];i;i=edge[i].next)
        if(!bz[edge[i].to])dfs1(edge[i].to,z);
}
void prepare(){
    int i;ia=0;
    fo(i,1,n)col[i]=-1;
    fo(i,1,n)if(col[i]==-1){
        cnt++;
        c[cnt].x=i;
        col[i]=0;
        dfs(i);
        if(ia==1){
            printf("-1");
            return;
        }
    }
}
int main(){
    scanf("%d",&n);
    fo(i,1,n)
        while(1){
            scanf("%d",&x);
            if(!x)break;
            map[i][x]=1;
        }
    fo(i,1,n)
        fo(j,1,i-1)
        if(!(map[i][j]&&map[j][i])){
            lb(i,j);lb(j,i);
        }
    prepare();
    if(ia)return 0;
    f[0][0]=1;
    fo(i,1,cnt){
        fo(j,c[i].g0,n)
            if(f[i-1][j-c[i].g0]){
                f[i][j]=1;
                g[i][j].x=i-1;
                g[i][j].y=j-c[i].g0;
                g[i][j].z=f[i-1][j-c[i].g0];
            }
        fo(j,c[i].g1,n)
            if(f[i-1][j-c[i].g1]){
                f[i][j]=2;
                g[i][j].x=i-1;
                g[i][j].y=j-c[i].g1;
                g[i][j].z=f[i-1][j-c[i].g1];
            }
    }
    fd(i,n/2,0)if(f[cnt][i]){
        x=cnt;y=i;z=f[cnt][i];
        dfs1(c[x].x,z-1);
        while(g[x][y].x||g[x][y].y){ 
            x2=g[x][y].x;
            y2=g[x][y].y;
            z=g[x][y].z;
            x=x2,y=y2;
            dfs1(c[x].x,z-1);
        }
        break;
    }
    fo(i,0,a0[0])printf("%d ",a0[i]);putchar('\n');
    fo(i,0,a1[0])printf("%d ",a1[i]);
    return 0;
}