poj 1182 食物链（DSU）

①普通并查集：

f[i]=i,表示i属于A

f[i]=i+n,表示i属于B

f[i]=i+2*n,表示i属于C

每次合并之前先进行判断

如果x和y属于同类，则需要判断(x,y+n)和(x,y+2*n)

如果x吃y，则需要判断(x,y)和(x+2*n,y)

②带权并查集：

每次合并压缩路径的时候， 模三取余，然后判断权值即可

附上普通并查集的代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=50005;
int n,k;
int f[3*maxn];

void Make_Set(){
	int Max=3*n;
	for(int i=0;i<Max;i++){
		f[i]=i;
	}
}

int find(int x){
	return x==f[x]?x:f[x]=find(f[x]);
}

void Union(int x,int y){
	int rx=find(x);
	int ry=find(y);
	f[ry]=f[rx];
}

bool judge(int x,int y){
	return find(x)==find(y);
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
	scanf("%d%d",&n,&k);
	Make_Set();
	int res=0;
	int op,x,y;
	while(k--){
		scanf("%d%d%d",&op,&x,&y);
		x--,y--;
		if(x<0||x>=n||y<0||y>=n) {res++;continue;}
		if(op==1){
			if(judge(x,y+n)||judge(x,y+2*n))
				res++;
			else{
				Union(x,y);
				Union(x+n,y+n);
				Union(x+2*n,y+2*n);
			}
		}
		else{
			if(judge(x,y)||judge(x,y+2*n))
				res++;
			else{
				Union(x,y+n);
				Union(x+n,y+2*n);
				Union(x+2*n,y);
			}
		}
	}
	printf("%d\n",res);
	return 0;
}