POJ 1182 食物链（并查集）

食物链

Description

动物王国中有三类动物A,B,C，这三类动物的食物链构成了有趣的环形。A吃B， B吃C，C吃A。 
现有N个动物，以1－N编号。每个动物都是A,B,C中的一种，但是我们并不知道它到底是哪一种。 
有人用两种说法对这N个动物所构成的食物链关系进行描述： 
第一种说法是"1 X Y"，表示X和Y是同类。 
第二种说法是"2 X Y"，表示X吃Y。 
此人对N个动物，用上述两种说法，一句接一句地说出K句话，这K句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。 
1） 当前的话与前面的某些真的话冲突，就是假话； 
2） 当前的话中X或Y比N大，就是假话； 
3） 当前的话表示X吃X，就是假话。 
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。 

Input

第一行是两个整数N和K，以一个空格分隔。 
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。 
若D=1，则表示X和Y是同类。 
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3

题目大意：有N只动物，编号为1,2,...,N。所有动物都属于A,B,C中的一种，已知A吃B，B吃C，C吃A。按顺序给出下面两种信息共K条。

第一种：x和y属于同一类

第二种：x吃y

然而这些信息有可能出错。有可能有的信息和之前给出的信息矛盾，也有的信息可能给出的x和y不在1至N的范围内。求在K条信息中有多少条是不正确的。

解题思路来自于《挑战程序设计竞赛》，看了半天终于看懂了。。。

解题思路：对于每只动物i创建3个元素i-A,i-B,i-C，并用这3 * N个元素建立并查集。其中1到N表示A类，N + 1到2N表示B类，2N + 1到3N表示C类。这个并查集维护如下信息：

i-X表示“i属于种类X”

并查集里的每一个组表示组内元素代表的情况都同时发生或不发生

因此，对于每条信息，只需按照下面的操作进行就可以了。

对于第一种信息，合并x-A,y-A或x-B,y-B或x-C,y-C。

对于第二种信息，合并x-A,y-B或x-B,y-C或x-C,y-A

在合并之前，要先检查是否有矛盾的信息出现。例如：在第一种信息的情况下，要检查x-A和y-B或y-C是否在同一组；在第二种信息的情况下，要检查x-A和y-A或y-C是否在同一组。

代码如下：

#include <cstdio>

const int maxk = 100005;
const int maxn = 50005;

int N,K;
int info[maxk],X[maxk],Y[maxk];
int par[3 * maxn],rank[3 * maxn];

void init(int n)
{
	for(int i = 1;i <= n;i++){
		par[i] = i;
		rank[i] = 0;
	}
}

int find(int x)
{
	return par[x] == x ? x : par[x] = find(par[x]);
}

void unite(int x,int y)
{
	x = find(x);
	y = find(y);
	if(x == y)
		return ;
	if(rank[x] < rank[y])
		par[x] = y;
	else{
		par[y] = x;
		if(rank[x] == rank[y])
			rank[x]++;
	}
}

bool same(int x,int y)
{
	return find(x) == find(y);
}

int main()
{
	scanf("%d %d",&N,&K);
	for(int i = 1;i <= K;i++)
		scanf("%d %d %d",&info[i],&X[i],&Y[i]);
	init(N * 3);
	int ans = 0;
	for(int i = 1;i <= K;i++){
		int x = X[i];
		int y = Y[i];
		
		if(x < 1 || N < x || y < 1 || N < y){
			ans++;
			continue;
		}
		
		if(info[i] == 1){//第一种信息 
			if(same(x,y + N) || same(x,y + 2 * N)){
				ans++;
			}else{
				unite(x,y);
				unite(x + N,y + N);
				unite(x + 2 * N,y + 2 * N);
			}
		}else{//第二种信息 
			if(same(x,y) || same(x,y + 2 * N)){
				ans++;
			}else{
				unite(x,y + N);
				unite(x + N,y + 2 * N);
				unite(x + 2 * N,y);
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}