题解 P1285 【队员分组】

· · 题解

思路:

具体:

f[i][j]表示的意思是,使用前i个联通块里的0或是1的染色点是否能达到j的容量。

f[i][j]=f[i][j-num[i][0]],j>=num[i][0]

f[i][j]=f[i][j-num[i][1]],j>=num[i][1]

数一下联通块个数,当作是01背包的个数。取染色为0的或者染色为1的背包都是可行的。记录路径pre[i][j]。结果要在f[blo_sum][j]找j(1<=j<=2/n)最大的一个。

无解情况:在染色的时候判断是否矛盾,即非二分图。

其他:

#include <bits/stdc++.h>
using namespace std;
const int MN=105;
const int MM=105;
typedef long long ll;

bool g[MN][MN],f[MN][MN];
int n,num[MM][2],pre[MN][MN],ans;
vector<int> cont[MN][2];

int mtc[MN],sum=0;
void dfs(int x,int fa,int col)
{
    num[sum][mtc[x]=col]++;cont[sum][col].push_back(x);
    for (int i=1;i<=n;++i)
        if (!g[i][x] && i!=fa && x!=i)
        {
            if (mtc[i]==-1) dfs(i,x,col^1);
            else if (mtc[i]==col) {printf("No solution");exit(0);}
        }
}
void dp()
{
    f[0][0]=true; //as a beginner.
    for (int i=1;i<=sum;++i)
        for (int j=0;j<=n/2;++j)
        {
            int t=j-num[i][0];
            //what can be carried must be the best one.
            if (t>=0&&f[i-1][t]) f[i][j]=true,pre[i][j]=0;
            t=j-num[i][1];
            if (t>=0&&f[i-1][t]) f[i][j]=true,pre[i][j]=1;
        }
    for (int i=n/2;i>=0;--i) if (f[sum][i]) {ans=i;break;}
}
void print()
{
    int res[MN]={0},t=ans;
    bool p[MN]={false};
    for (int i=sum;i>0;--i)
    {
        for (int j=0;j<cont[i][pre[i][t]].size();++j) p[res[++res[0]]=cont[i][pre[i][t]][j]]=true;
        t-=num[i][pre[i][t]];
    }
    sort(res+1,res+res[0]+1);
    printf("%d ",res[0]);for (int i=1;i<=res[0];++i) printf("%d ",res[i]);
    printf("\n%d ",n-res[0]);for (int i=1;i<=n;++i) if (!p[i]) printf("%d ",i);
}
ll read(){
    ll f=1,x=0;char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c<='9'&&c>='0'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int main()
{
    n=read();
    int m,v;
    for (int i=1;i<=n;++i)
    {
        v=-1;
        while (v!=0){v=read();g[i][v]=true;}
    }
    for (int i=1;i<=n;++i)
        for (int j=i+1;j<=n;++j)
        {
            if (!(g[i][j]&&g[j][i]) && (g[i][j]||g[j][i])) g[i][j]=g[j][i]=false;
        }
    memset(mtc,-1,sizeof(mtc));
    for (int i=1;i<=n;++i)
        if (mtc[i]==-1)
        {
            sum++;
            dfs(i,0,1);
        }
    dp();
    print();
    return 0;
}