poj1182：带权并查集

poj1182

经典题目

题解：

对于k,x,y
k=0,1,2分别表示同类，x吃y，x被y吃
如果x吃y，y吃z，则z吃x，所以k分别为1，1，2
如果x被y吃，y被z吃，则x吃z，k分别为2，2，1
如果x和y同类，y吃z，则x吃z，k分别为0，1，1
如果x和y同类，y被z吃，则x被z吃，k分别为0，2，2
我们发现（k1 + k2） % 3 = k3

代码：

#include <cstring>
#include <algorithm>
#include <cstdio>
#include <iostream>
using namespace std;
int const N = 50000 + 10;
int n,k,fa[N],re[N];
int find(int x){
	if(x != fa[x]){
		int tmp = fa[x];
		fa[x] = find(fa[x]);
		re[x] = (re[x] + re[tmp]) % 3;
	}
	return fa[x];
}
int main(){   //re数组初始化为0，因为它的父亲是自己，自己和自己是同类
	cin>>n>>k;
	int d,x,y,ans = 0;
	for(int i=1;i<=n;i++)	fa[i] = i;
	while(k--){
		scanf("%d%d%d",&d,&x,&y);
		if(x > n || y > n || (d == 2 && x == y))	++ans;
		else{
			int px = find(x),	py = find(y);
			if(px == py){
				if((re[x]-re[y]+3)%3 != (d-1))	++ans;
			}else{
				fa[px] = py;
				re[px] = (d-1+re[y]-re[x])%3;
			}
		}
	}
	cout<<ans;
	return 0;
}