POJ - 1417(True Liars)

题意：雾岛上有两个部落，天使部落有 p1 个人，他们只说真话，恶魔部落有 p2 个人，他们只说假话，但是两个部落的人长得一模一样，把这 p1+p2 个人从1编号，然后有 n 次询问，每次询问给出 a b s ，表示你询问编号为a的人b是否属于天使部落，回答是则s为“yes” ，回答否则“no”，(注意a的部落决定他说的是真话还是假话)，求根据这 n 次询问能否推断出哪 p1 个人是天使部落的人，不能则输出no，否则给出所有天使部落人的编号；

 

分析：

Ⅰ：先分析查询，有四种情况；

① a属于天使部落，b属于天使部落，s="yes";

② a属于天使部落，b属于恶魔部落，s="no";

③ a属于恶魔部落，b属于天使部落，s="no";

④ a属于恶魔部落，b属于恶魔部落，s="yes";

可以总结出，只要 a 和 b 是同一部落，s就是yes，否则就是no，所以可以想到带权种类并查集；

Ⅱ：根据种类并查集的性质，在上述 n 个询问操作之后，所有 p1+p2 个人被分成若干个连通块，每个连通块根据种类分成两类，一个连通块的两类肯定分别属于两个部落，只是我们不能确定具体属于。然后题目要求判断是否可以推断出哪 p1 个人是天使部落的人，即在上述 n 个询问后，只有一种合法的组合方式可以组成 p1 个人，这就需要DP：

我们统计连通块的数量并编号，统计对应联通块两类的数量，这样应该很好想到可以用DP，设一共有cnt个连通块，设dp[i][j] 表示用前i个连通块组成数量为 j 的方案数，那么只有最后 dp[cnt][p1] = 1 时，组合方式才是唯一的，再开个对应数组记录路径即可；

 

 

代码：

#include<map>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

const int N = 600+10;
int fa[N],v[N],n,m,p1,p2;
int path[N][N];
int  ant[N][2];
int   dp[N][N];
bool ans[N][2];

void init(){
	n=p1+p2;
	for(int i=0;i<=n;i++){
		fa[i]=i;
		 v[i]=0; 
	}
	memset(dp,0,sizeof(dp));
	memset(ant,0,sizeof(ant));
	memset(path,0,sizeof(path));
	memset(ans,0,sizeof(ans));
}

int find(int x){
	if(x==fa[x]) return x;
	int root=find(fa[x]);
	v[x]=(v[x]+v[fa[x]])%2;
	return fa[x]=root; 
}

void merge(int a,int b,int op){   //种类并查集写法不唯一，看个人
	int x=find(a),y=find(b);
	fa[x]=y;
	 v[x]=(v[b]-v[a]+2+op)%2;
}

int main()
{
	while(~scanf("%d%d%d",&m,&p1,&p2)&&m+p1+p2){
		init();
		while(m--){
			int a,b;char s[3];
			scanf("%d%d",&a,&b);
			scanf("%s",s);
			if(s[0]=='y'){
				merge(a,b,0);
			}
			else{
				merge(a,b,1);
			}
		}
		map<int,int>mp; int cnt=0;
		for(int i=1;i<=n;i++){    //统计连通块的数量并编号
			if(find(i)==i){
				mp[i]=++cnt; 
			}
		}
		for(int i=1;i<=n;i++){
			int f=find(i);
			ant[mp[f]][v[i]]++;   //统计每个连通块两类的数量
		}
		dp[0][0]=1;
		for(int i=1;i<=cnt;i++){
			for(int j=p1;j>=1;j--){
				if(j>=ant[i][0]&&dp[i-1][j-ant[i][0]]){
					dp[i][j]+=dp[i-1][j-ant[i][0]]; path[i][j]=ant[i][0];  
				}
				if(j>=ant[i][1]&&dp[i-1][j-ant[i][1]]){
					dp[i][j]+=dp[i-1][j-ant[i][1]]; path[i][j]=ant[i][1];
				} 
			}
		}
		if(dp[cnt][p1]>1){
			puts("no");
		}
		else{
			int k=p1;
			for(int i=cnt;i>=1;i--){
				int op=path[i][k];
				if(op==ant[i][0]){
					ans[i][0]=true;
				}
				else{
					ans[i][1]=true;
				}
				k-=op;
			}
			for(int i=1;i<=n;i++){
				int f=find(i);
				if(ans[mp[f]][v[i]]){
					printf("%d\n",i);
				}
			}
			puts("end");
		}
	}
}