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

题目分析

这道题需要我们维护并查集的同时维护每种动物与其它动物的关系（同类，吃，被吃）。本题的难点也就是如何维护这个关系。
我们可以通过维护每个节点到其根节点的距离，进而来表示每个动物之间的关系。

设x与其根节点px。d[x]表示x到根节点的距离。
d[x]%3=0 //表示x与px是同类
d[x]%3=1 //表示px吃x
d[x]%3=2 //表示x吃px

这样假设完之后，我们可以对其进行推广：
对于位于同一集合中的任意两点x、y，如果：
(d[x]-d[y])%3==0，则x与y是同类
(d[x]-d[y])%3==1，则x吃y
(d[x]-d[y])%3==2，则y吃x

然后我们再来看看上面三种不合法的情况。

1. 当前的话与前面的某些真的话冲突，就是假话；
   与前面的真话发生冲突：即不符合上面推广的三种情况。在操作时加一个特判即可。
2. 当前的话中X或Y比N大，就是假话；
   这个直接判断即可，没什么可说的。
3. 当前的话表示X吃X，就是假话。
   (d[x]-d[x])%3一定是等于0的，判断也符合上述的推广，利用推广条件进行判断即可。

这道题的最后一个难点是d[]的维护。
d[x] //表示x到其父节点p[x]的权值
注：因为在操作前，所有节点都先要调用find()函数找到其根节点。通过路径压缩，x的父节点p[x]会变为根节点。即：d[x]在实际操作中可以表示x到其根节点的权值。

1. find()+路径压缩时维护d[]：因为递归调用+路径压缩，p[x]的父节点一定是根节点。所以d[p[x]]为父节点到根节点的权值，d[x]为x到父节点的权值。将x连接到根节点上，d[x]变为x到根节点的权值=原(d[p[x]]+d[x])。
2. 合并时d[]的维护：将以pb为根的集合接到pa上，那么我们只需要更新d[pb]的值即可。设a和b分别为pa和pb集合内的一点，dist(x)表示x到根节点的权值。
   以a与b的关系是同类为例：
   因为a和b是同类，因此：(dist(a)-dist(b))%3==0，即(d[a]-d[b]-d[pb])%3==0 => d[pb]=d[a]-d[b]。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#include <iomanip>
#define LL long long
#define PII pair<int,int>
using namespace std;
const int N=1e5+5,INF=0x3f3f3f3f;
int p[N],d[N]; 
int find(int x)			//找到x的根节点+路径压缩优化
{
	if(p[x]!=x)
	{
		int t=find(p[x]);		//递归求出p[x]
		d[x]+=d[p[x]];			//更新d[x]
		p[x]=t;					//更新p[x]
	}
	return p[x];
}
int main()
{
	int n,m;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++) p[i]=i;
	int ans=0;
	while(m--)
	{
		int op,x,y;
		scanf("%d %d %d",&op,&x,&y);
		if(x>n||y>n) ans++;			//条件2
		else
		{
			int px=find(x),py=find(y);		//先求出x和y的根节点，并且更新了d[x]与d[y]的值
			if(op==1)
			{
				if(px==py&&(d[x]-d[y])%3) ans++;	//利用推广条件进行判断
				else if(px!=py)						//合并集合
				{
					p[py]=px;
					d[py]=d[x]-d[y];
				}
			}
			else 
			{
				if(px==py&&(d[x]-d[y]-1)%3) ans++;
				else if(px!=py)
				{
					p[py]=px;
					d[py]=d[x]-d[y]-1;
				}
			}
		}
	}
	printf("%d\n",ans); 
	return 0;
}