POJ 1417 True Liars(路径压缩并查集+DP背包问题)

POJ 1417 True Liars(路径压缩并查集+DP背包问题)

http://poj.org/problem?id=1417

题意：

        给出p1+p2个人，其中p1个是好人，p2个是坏人。然后有一些关系 ，a说b是好人(坏人).其中没有矛盾的，判断是否有唯一解判断哪些人是好人，哪些人是坏人。

        其中比较重要的是，好人总说真话，坏人总说假话(这句话的特殊意义)。不会存在矛盾情况。请问你是否存在唯一解，如果存在请输出唯一解。

分析：

        如果A说B是好人，那么A与B是同一类人。如果A说B是坏人，那么A与B不同类。(想想为什么是这样)

        所以原题所给的每句话就是一条路径压缩并查集的关系，那么我们最终合并所有关系可以得到cnt个连通分量且每个分量中节点的相互关系(同类or异类)我们都知道。

        好，现在假设有cnt个连通分量，第一个分量有x1与y1个两类人(我们不知道到底x1那些人是好人还是y1那些人是好人)，第二个分量有x2与y2个两类人...所以我们现在想知道是否只有一种方式让我们从第一分量中抓X1(或Y1)个人来，从第二分量中抓X2(或Y2)人来...直到每个分量都抓一类人时，正好抓了p1个人。

        如果只有1种方式实现上面的目的，那么就有唯一解。记录DP的每次选择最后输出即可。

        令d[i][j]表示取完前i个分量后正好j人的方法数，我们最后要求的是看d[cnt][p1]是否==1?

        DP转移方程:dp[i][j]=sum{ dp[i-1][j-bag[i][0]]+dp[i-1][j-bag[i][1]] }

 AC代码：

<span style="font-size:18px;">#include<cstdio>
#include<cstring>
#include<map>
using namespace std;
const int MAXN=1000;
int bag[MAXN][2];//bag[i][0]表示第i(重新编号了)个连通分量的0类人的个数,bag[i][1]表1类人
map<int,int> mp;//用来将老的连通分量编号映射bag中的新编号
int cnt;//一共有cnt个分量
int insert(int x,int b)//连通分量x,与x的关系是b(0表示同类,1表示异类)
{
    if(mp.find(x)==mp.end())mp[x]=++cnt;
    bag[mp[x]][b]++;//该分量的b类人加1
    return mp[x];
}
int F[MAXN];
int v[MAXN];//表示i与根的关系
int findset(int i)
{
    if(F[i]==-1)return i;
    int temp= findset(F[i]);
    v[i] =(v[i]+v[F[i]])%2;
    return F[i]=temp;
}
void bind(int i,int j,int temp)
{
    int fa=findset(i);
    int fb=findset(j);
    if(fa!=fb)
    {
        F[fb]=fa;
        v[fb]=(v[i]+v[j]+temp)%2;
    }
}
int d[MAXN][310];//DP
int main()
{
    int n,p1,p2;
    while(scanf("%d%d%d",&n,&p1,&p2)==3)
    {
        if(n==0&&p1==0&&p2==0)break;
        cnt=0;
        mp.clear();
        memset(bag,0,sizeof(bag));
        memset(F,-1,sizeof(F));
        memset(v,0,sizeof(v));
        memset(d,0,sizeof(d));
        while(n--)
        {
            int a,b,temp;
            char str[10];
            scanf("%d%d%s",&a,&b,str);
            if(str[0]=='y')
                temp=0;
            else if(str[0]=='n')
                temp=1;
            int fa=findset(a);
            int fb=findset(b);
            if(fa!=fb)//不同分量
                bind(a,b,temp);;
        }
        for(int i=1;i<=p1+p2;i++)//将1到p1+p2所有点重新编号
        {
            int fi=findset(i);
            insert(fi,v[i]);
        }
        d[0][0]=1;//初值
        for(int i=1;i<=cnt;i++)//连通分量编号从1到cnt
        {
            for(int j=0;j<=p1;j++)
            {
                if( j>=bag[i][0] )
                    d[i][j] = d[i-1][j-bag[i][0]];
                if( j>=bag[i][1] )
                    d[i][j] += d[i-1][ j-bag[i][1] ];
            }
        }
        //printf("###%d\n",d[cnt][p1]);

        if(d[cnt][p1]==1)//能区分出
        {
            int j=p1;
            int choose[MAXN];//choose[i]=1/0表示第i(重新编号)个连通分量选择第0类还是选第1类
            memset(choose,-1,sizeof(choose));
            for(int k=cnt;k>=1;k--)//逆推找出choose
            {
                if( d[k][j] == d[k-1][j-bag[k][0]] )
                {
                    choose[k]=0;
                    j=j-bag[k][0];
                }
                else if( d[k][j] == d[k-1][j-bag[k][1]] )
                {
                    choose[k]=1;
                    j=j-bag[k][1];
                }
            }
            for(int i=1;i<=p1+p2;i++)
            {
                int fa=findset(i);//找出分量的编号fa
                int num=mp[fa];//找出fa重新编号后的编号 num
                if(v[i]==choose[num])
                    printf("%d\n",i);
            }
            printf("end\n");
        }
        else
        {
            printf("no\n");
        }

    }
    return 0;
}
</span>