Gym - 101128B[并查集]

题目链接:http://fastvj.rainng.com/problem/Gym-101128B

 

解题思路:

其实这是一个sat问题，当我们枚举要拿那三张牌的时候，如果条件中的两张牌在23张里面，并且是一人分一张的情况，那么此时我们就无法直接判断出牌的归属。

那么此时不妨设一个并查集，将不能跟牌i在一起的都归并带f[i+26]当中，正所谓敌人的敌人是朋友，只有两种情况，要么都不在一起，要么都在一起，所以完全可以并查集。

然后还需要注意的一点是，当我们确定了牌i的归属时，也就确定了i的"敌人集"和"朋友集"的归属，此时就要更新并查集。所以并查集内的点肯定是不确定归属的点。

#include<math.h>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
typedef long long ll;
const int mx = 100;
int n,fa[mx],be[mx];
struct node
{
	char c[5];
	int w,no;
}s[mx];
int find(int x){
	return x==fa[x]? x:fa[x]=find(fa[x]);
}
bool in(int A,int B,int C,int x){
	return A==x||B==x||C==x;
}
void full(int w,int x){
	for(int i=0;i<26;i++)
	if(find(i)==x) be[i] = w,fa[i] = i;
}
bool check(int A,int B,int C)
{
	for(int i=0;i<60;i++) fa[i] = i;
	memset(be,-1,sizeof(be));
	be[A] = be[B] = be[C] = 2;
	for(int i=1;i<=n;i++){
		int c1 = s[i].c[0] - 'A',d = s[i].no;
		int c2 = s[i].c[1] - 'A',w = s[i].w;
		bool in1 = in(A,B,C,c1),in2 = in(A,B,C,c2);
		if(!d){
			if(be[c1]==w||be[c2]==w) return 0;
			if(!in1) be[c1] = w^1,full(w,find(c1+26)),full(w^1,find(c1));
			if(!in2) be[c2] = w^1,full(w,find(c2+26)),full(w^1,find(c2));
		}else if(d==1){
			if(in1&&in2) return 0;
			if(in1){
				if(be[c2]==(w^1)) return 0;
				be[c2] = w;
				full(w^1,find(c2+26));
				full(w,find(c2));
			}else if(in2){
				if(be[c1]==(w^1)) return 0;
				be[c1] = w;
				full(w^1,find(c1+26));
				full(w,find(c1));
			}else{
				if(be[c1]==be[c2]&&be[c1]!=-1) return 0;
				if(be[c1]!=-1&&be[c2]==-1){
					be[c2] = be[c1]^1;
					full(be[c2],find(c2));
					full(be[c1],find(c2+26));
					continue; 
				}
				if(be[c2]!=-1&&be[c1]==-1){
					be[c1] = be[c2]^1;
					full(be[c1],find(c1));
					full(be[c2],find(c1+26));
					continue;
				}
				int x = find(c1),y = find(c2);
				if(x==y) return 0;
				fa[x] = find(c2+26);
				fa[y] = find(c1+26);
			}
		}else if(d==2){
			if(in1||in2) return 0;
			if(be[c1]==(w^1)) return 0;
			if(be[c2]==(w^1)) return 0;
			int x = find(c1),y = find(c2);
			if(x==find(c2+26)||y==find(c1+26)) return 0;
			full(w^1,find(c2+26));
			full(w^1,find(c1+26));
			full(w,find(c1));full(w,find(c2));
			be[c1] = be[c2] = w;
		}
	}
	return 1;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%s%d%d",s[i].c,&s[i].w,&s[i].no);
		s[i].w--;
	}
	int ans = 0;
	for(int i=0;i<26;i++){
		for(int j=i+1;j<26;j++){
			for(int k=j+1;k<26;k++)
			if(check(i,j,k)) ans++;
		}
	} 
	printf("%d\n",ans);
	return 0;
}