POJ1182:食物链 （并查集）（构建三个节点）（巧妙）

1182:食物链

• 查看
• 提交

总时间限制: 1000ms  内存限制: 65536kB

描述

动物王国中有三类动物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），输出假话的总数。

输入

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

输出

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

样例输入

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

样例输出

3

【思路】看了一个大神的思路，非常巧妙，容易理解，大概是给每个动物三个节点，用这三个节点的并查集来确定两动物之间的关系，原文链接：https://blog.csdn.net/lisong_jerry/article/details/80029967，在此简单阐述一下大神的思路

【构建】数组tree[N*3]，表示N*3个点的父节点。如下图，动物a、b分别用三个节点表示，a的三个节点的父节点为tree[a] tree[a+N] tree[a+2*N]

1.同类关系的表述：tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N]）

2.吃与被吃关系的表述：tree[a]==tree[b+N],tree[a+N]==tree[b+2N],tree[a+2*N]==tree[b]）

3.（第一种ab同类关系）a吃c，b吃c：查找root时发现，tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N]）

4.（第二种ab同类关系）a吃c c吃d d吃b关系：（查找root时发现，tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N]）

【综上所述】

如果ab同类，则必有tree[a]==tree[b],tree[a+N]==tree[b+N],tree[a+2*N]==tree[b+2*N]），如果a吃b，则必有tree[a]==tree[b+N],tree[a+N]==tree[b+2N],tree[a+2*N]==tree[b]），根据如上关系可以编写代码：

#include<iostream>
#include<cstring>
#include<vector>
#include<math.h>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
#define maxi 0x3f3f3f3f
int N,K;
int tree[200000];// 3*N
int findroot(int x)
{
	if(tree[x]==x)
		return x;
	else
	{
		int temp=findroot(tree[x]);
		tree[x]=temp;
		return temp;
	}
}
bool same(int x,int y)
{
	if(findroot(x)==findroot(y))return true;
	return false;
}
void unite(int x,int y)
{
	int fx=findroot(x);
	int fy=findroot(y);
	if(fx!=fy)  // 忘掉了，应该是把他们的root连通，而不是让tree[x]=y
		tree[fx]=fy;
}
int main(int argc, char const *argv[])
{
	cin>>N>>K;
	int ans=0;
	int d,x,y;
	for(int i=1;i<200000;i++)tree[i]=i;
	while(K--)
	{
		cin>>d>>x>>y;
		if(x>N||y>N||x<1||y<1)ans++;
		else if(d==2&&x==y)ans++;
		else if(d==1)
		{
			if(same(x,y+N)||same(y,x+N))ans++; //如果是x吃y或者y吃x的关系 直接谎话+1
			else
			{
				unite(x,y);
				unite(x+N,y+N);
				unite(x+2*N,y+2*N);
				
			}
		}
		else// d==2
		{
			if(same(x,y)||same(y,x+N))ans++; // 如果是同类或者y吃x 直接谎话+1
			else
			{
				unite(x,y+N);
				unite(x+N,y+2*N);
				unite(x+2*N,y);
				
			}

		}
	}
	cout<<ans<<endl;
	return 0;
}