食物链（并查集）

题目链接

分析可知，只要知道x和z、y和z两对关系，就能知道x、y、z两两之间的关系。

用一棵树表示节点之间的关系，对于任意两个节点，只要知道每个节点和根节点之间的关系， 就可知道这两个节点之间的关系了。

用 “每个节点到根节点的距离”来表示“每个节点与根节点的关系”，用并查集维护 “每个节点到根节点的距离”这一信息。

设根节点为z，d[i]表示节点i到根节点的距离，分析可知，在模3意义下，如果d[i]-d[z]==1，则i吃z，如果d[i]-d[z]==2，则z吃i，如果d[i]-d[z]==0，则i和z是同类。

假设需要查询关系的两个点为x和y，在模3意义下，如果d[x]==d[y]，则x和y是同类（x和y到根节点的距离相等），如果d[x]==d[y]+1，则x吃y（x到根节点的距离比y到根节点的距离多1）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005;
int p[N],d[N];
//d[x]要更新成x到根节点的距离 x到根节点的距离=x到父节点的距离+x的父亲到根节点的距离 d[x]+d[p[x]] 递归可算
int find(int x){
	if(p[x]!=x) { 
		int t=find(p[x]);//递归寻找p[x]的根节点t
		d[x]+=d[p[x]];// 回溯中，d[p[x]]现在存的是x的父亲到根节点的距离 
		p[x]=t;//路径压缩
	}
	return p[x];
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i)p[i]=i;
	int res=0;
	while(k--){
		int t,x,y;
		scanf("%d%d%d",&t,&x,&y);
		if(x>n||y>n) res++;
		else{
			int px=find(x),py=find(y);
			if(t==1){
				if(px==py&&(d[x]-d[y])%3)res++;
				else if(px!=py){//x和y不在一棵树上，进行合并
					p[px]=py;//px认py作父
					d[px]=d[y]-d[x];//d[x]+d[px]==d[y](%3) x与y同类
				}
			}else{
				if(px==py&&(d[x]-d[y]-1)%3)res++;
				else if(px!=py){
					p[px]=py;
					d[px]=d[y]+1-d[x];//d[x]+d[px]==d[y]+1(%3) x吃y
				}
			}
		}
	}
	printf("%d\n",res);
}