1. 程式人生 > >True Liars[擴充套件域並查集]

True Liars[擴充套件域並查集]

傳送門

每個點拆成兩個,表示好人或壞人

我們合併集合後,發現存在幾組對立的集合

也就是說這個集合和與它對立的集合只能選一個

我們用rt1[i] , rt2[i] 表示第i個集合 和 與第i個集合對立的集合

cnt1,cnt2表示該集合好人的個數

用f[i][j]表示到第i個集合,好人為j的方案數

f[i][j]+=f[i-1][j-cnt1[i]] (f[i-1][cnt1[i-1]!=0)

f[i][j]+=f[i-1][j-cnt2[i]](f[i-1][j-cnt2[i]!=0)

同時記錄from[i][j] 表示f[i][j]選的是第i組集合的哪一個集合


#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 305*2*2
using namespace std;
int fa[N],n,p1,p2,rt1[N],rt2[N],tot;
int cnt1[N],cnt2[N],f[N/2][N/2],from[N/2][N/2];
int pre[N],vis[N],ans[N];
void dfs(int i,int j){
	if(i==0 && j==0) return;
	if(from[i][j]==1) vis[pre[i]]=1,dfs(i-1,j-cnt1[i]);
	else vis[pre[i]+p1+p2]=1,dfs(i-1,j-cnt2[i]);
}
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void init(){
	memset(f,0,sizeof(f));
	memset(from,0,sizeof(from));
	memset(vis,0,sizeof(vis));
	memset(ans,0,sizeof(ans));
	memset(cnt1,0,sizeof(cnt1));
	memset(cnt2,0,sizeof(cnt2));
	memset(rt1,0,sizeof(rt1)); tot=0;
	memset(rt2,0,sizeof(rt2));
}
int main(){
	while(scanf("%d%d%d",&n,&p1,&p2) && (n+p1+p2)){
		init();
		for(int i=1;i<=2*(p1+p2);i++) fa[i]=i;
		for(int i=1;i<=n;i++){
			int x,y,x1,y1; char s[5];
			scanf("%d%d%s",&x,&y,s);
			x1=find(x+p1+p2),y1=find(y+p1+p2),x=find(x),y=find(y);
			if(s[0]=='y') {
				if(x!=y) fa[x]=y;
				if(x1!=y1) fa[x1]=y1;
			}
			if(s[0]=='n'){
				if(x!=y1) fa[x]=y1;
				if(x1!=y) fa[x1]=y;
			}
		}
		for(int i=1;i<=p1+p2;i++){
			if(find(i)==i) rt1[i]=++tot,pre[tot]=i;
		}
		for(int i=1;i<=tot;i++) rt2[pre[i]+p1+p2]=i;
		for(int i=1;i<=p1+p2;i++){
			cnt1[rt1[find(i)]]++ , cnt2[rt2[find(i)]]++;
		}
		f[0][0]=1;
		for(int i=1;i<=tot;i++)
			for(int j=min(cnt1[i],cnt2[i]);j<=p1;j++){ 
				if(f[i-1][j-cnt1[i]]) f[i][j]+=f[i-1][j-cnt1[i]] , from[i][j]=1;
				if(f[i-1][j-cnt2[i]]) f[i][j]+=f[i-1][j-cnt2[i]] , from[i][j]=2; 
			}
		if(f[tot][p1]!=1) {printf("no\n"); continue;}
		dfs(tot,p1);
		for(int i=1;i<=tot;i++) {
			if(vis[pre[i]]) for(int j=1;j<=p1+p2;j++) if(find(j)==pre[i]) ans[j]=1;
			if(vis[pre[i]+p1+p2]) for(int j=1;j<=p1+p2;j++) if(find(j)==pre[i]+p1+p2) ans[j]=1;
		}
		for(int i=1;i<=p1+p2;i++) if(ans[i]) printf("%d\n",i);
		printf("end\n");
	}return 0;
}