食物链（并查集）

题目：

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

输入格式

第一行是两个整数 N和 K，以一个空格分隔。

以下 K 行每行是三个正整数 D，X，Y两数之间用一个空格隔开，其中 D 表示说法的种类。

若 D=1，则表示 X和 Y是同类。

若 D=2，则表示 X 吃 Y。

输出格式

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

数据范围

1≤N≤50000
0≤K≤100000

输入样例：

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

输出样例：

3

分析：

该题的关键在于维护一个数组d[ ]，用于表示每个节点到根节点的距离；

假设有一条食物链 a—b—c—d—e—f—g   (其中a为根节点，并且b吃a)

那么d[a]=0     d[b]=1     d[c]=2     d[d]=3     d[e]=4     d[f]=5     d[g]=6

由于abcdefg必定为A，B，C中的一类，所以

由b吃a，c吃d，则有a吃c；因为d吃c，所以a，d为同类

同理，a，d，g为一类；b，e为一类；c，f为一类；

根据规律可知

若d[x]%3=0，则x和根节点是同类==>若(d[x]-d[y])%3=0，则x和y是同类

若d[x]%3=1，则x能够吃根节点同类==>若(d[x]-d[y])%3=1，则x能够吃y

若d[x]%3=2，则x会被根节点同类吃==>若(d[x]-d[y])%3=2，则x会被y吃

本题可以分为以下几种情况讨论

一、x>n或者y>n，那么当前的话为假，res++；

二、若x<=n且y<=n,那么

        1、t=1时；

                （1）若x，y在同一集合，则判断x，y是否为同类

                （2）若x，y在不同集合，则将x，y所在集合合并

        2、t=2时；

                （1）若x，y在同一集合，则判断x是否能吃掉y

                （2）若x，y在不同集合，则将x，y所在集合合并

代码如下：

#include<iostream>
using namespace std;
const  int N=100010;
int n,m;
int p[N],d[N];//p[N]寻找父节点，d[]求到祖宗节点的距离

int find(int x){            //用于查询x的根节点
	if(p[x]!=x)             //判断x是否为根节点
	{
		int t=find(p[x]);   //用t暂存p[x]的根节点 
		d[x]+=d[p[x]];      //d[x]一开始表示x到父节点的距离，d[p[x]]表示p[x]到根节点的距离 
		p[x]=t;
	}
	return p[x];            //返回x的根节点
} 

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++)p[i]=i; //表示初始状态时，每个节点的根节点是它本身
	int res=0;
	while(m--){
		int t,x,y;
		cin>>t>>x>>y;
		if(x>n||y>n)res++;
		else{
			int px=find(x),py=find(y);  //寻找祖宗节点
			if(t==1){                   //判断是否为同类 
				if(px==py){             //如果x，y在同一个集合中 
					if((d[x]-d[y])%3) res++;     //两数到根节点距离之差的模不为0，说明不是同类 
//	因为d[x]和d[y]可能是负数，C++里负数模正数还是负数。
//所以式子可以写为(d[x] % 3 + 3) % 3 != (d[y] % 3 + 3) % 3			
				}else{                  //如果x，y不在同一个集合中 
					p[px]=py;           //x所在集合并到y所在集合
					d[px]=d[y]-d[x]; 
//由于x->px的距离为d[x]
//    y->py的距离为d[y]
//把两个集合合并，可以将px指向py
//那么px->py的距离为d[px]   (py为祖宗节点）
//由于x与y同类，所以x与y到根节点py距离之差是3的倍数
//所以(d[x]+d[px]-d[y])%3=0
//所以可以设 d[px]=d[y]-d[x]   d[x]在find函数中更新 
				}
			}
			if(t==2){
				if(px==py){
					if((d[x]-1-d[y])%3)res++;
				}else{
					p[px]=py;
					d[px]=d[y]-d[x]+1;
//同理 d[x]+d[px]-d[y]=1 
				}
			} 
		}
	}
	cout<<res;
	return 0;
}