食物链-------带权并查集

食物链
动物王国中有三类动物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
思路：带权并查集
本题关系：…C->A->B->C->A…，形成一个由三个点组成的环状关系，那就根据题意为三个关系赋值，0代表同类，1代表A吃B，2代表A被B吃（主要和权值的计算有关，可以为任何值赋任何含义）。如果两个点的父节点相同，那就判断他们与父节点的关系，从而判断给出的值是否合理，如果两个节点的父节点不同，说明他们两个第一次产生关系，值合理
d[i]表示i到i的爹的权值

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 50010;
int f[N],d[N];
int n,p,ans;
int find (int a)	//再找爹的途中顺便记一下找爹所走的路程（权）
{
	if(f[a]!=a)
	{
		int p=f[a];
		f[a]=find(p);
		d[a]+=d[p]; 
	}
	return f[a];
}
int main()
{
	scanf("%d%d",&n,&p);
	for(int i=1;i<=n;i++)f[i]=i;	//日常初始化
	ans=0;
	while(p--)
	{
		int t,a,b,l;
		scanf("%d%d%d",&t,&a,&b);
		if(t==2&&a==b){ans++;continue;}
		if(a>n||b>n){ans++;continue;}		//判断的条件
		if(t==1)l=0;
		else	l=1;				//0表示同类，1表示吃
		int fa=find(a),fb=find(b);
		
		if(fa==fb)
		{
			if(((d[b]-d[a])%3+3)%3!=l)ans++;	//d[b]-d[a]为a，b通过爹连起来后的权值，
												//%3转化成我们定义的关系
										//为什么+3）%3，自己画一个三个点的环想一下就好了，说不清楚
		}
		else
		{
			f[fb]=fa;
			d[fb]=d[a]+l-d[b];			//更新b的爹，到b的爹的爹的权值
		}
	}
	cout<<ans<<endl;
}

GG