POJ-1182 食物链(并查集)

传送门

食物链

动物王国中有三类动物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

思路

这一题和POJ-1703 Find them, Catch them是同类型的题，都是利用并查集维护同类元素关系的问题，只不过1703元素间只有两种关系：朋友或敌人。而本题是三个元素形成的种间关系，1吃2,2吃3,3吃1。
思路是一样的，我们维护两种关系的元素时开辟了2xn个空间，那么三种关系就要开辟3*n个空间。
实际上对于元素a分析，a和其他元素间的关系为1.朋友、2.a吃ta、3.a被ta吃.所以我们模拟a吃a+n,a+n吃a+2xn,a+2xn吃a,即成功模拟出了这条食物链的关系。
接下来即开始判断：
1.a和b为同种：只要a不吃b或a不被b吃即与a、b同种不矛盾。
2.a捕食b：只要a和b不为同种或a不被b吃即与a捕食b不矛盾。

解决代码

#include<iostream>
using namespace std;
const int maxx=2e5+5;
int a[maxx];
int h[maxx];
void init(int n)
{
	for (int i = 1; i <= n; i++)
	{
		a[i]=i;
		h[i]=0;
	}
}
int find(int x)
{
	return x==a[x]? x:a[x]=find(a[x]);
}
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y) return ;
	if (h[x]<h[y])
	{
		a[x]=y;
	}
	else
	{
		a[y]=x;
		if(h[x]==h[y]) h[x]++;
	}
}
bool same(int x,int y)
{
	return find(x)==find(y);
}
int main()
{
	int tot=0;
	int n,m;
	scanf("%d%d",&n,&m);
	//cin>>n>>m;
	init(3*n);
	for (int i = 0; i < m; i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		//cin>>x>>y>>z;
		if(y<1||z<1||y>n||z>n)
		{
			tot++;
			continue;
		}
		if (x==2)
		{
			if(y==z)
			{
				tot++;
				continue;
			}
			if(same(y,z)||same(y,z+n)) tot++;
			else//没有同类或被捕食关系
			{
				unite(y+n,z);
				unite(y,z+2*n);
				unite(2*n+y,z+n);
			}
		}
		else//没有捕食或被捕食关系
		{
			if(same(z,y+2*n)||same(z,y+n)) tot++;
			else
			{
				unite(y,z);
				unite(y+n,z+n);
				unite(y+2*n,z+2*n);
			}
		}
	}
	printf("%d\n",tot);
	//cout<<tot<<endl;
	return 0;
}

ps：这道题用cin和cout会tle