poj 1182 食物链

题目是中文的题目意思就不多说了。这是本来是并查集但是，并不是简单的并查集。这是一题带权并查集。

基本的并查集套用基本的模板就行了。

int findx(int x)
{
    int r=x;
    while(bin[r] !=r)
        r=bin[r];
    return r;
}

 

这题要在查找的时候找到权值。带权并查集和普通并查集最大的区别在于带权并查集合并的是可以推算关系的点的集合

假设arr[a]=b,rank[a]=0表示同类；rank[a]=1表示b可以吃a；rank[a]=2表示a可以吃b。

对于一组数据d,x,y,若x与y的根节点相同，利用(rank[y]-rank[x]+3)%3!=d-1若不相等说明为假话；若x与y根节点不同，说明两者不在同一集合里，

进行组合。让x的根fx成为y的根fy的父节点（arr[fy]=fx）,rank[fy]的值需要仔细归纳。

#include<stdio.h>
#include<string.h>
int arr[50005];
int rank[50005];
void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		arr[i]=i;
	}
	memset(rank,0,sizeof(rank));
}
int find(int x)
{
	if(x!=arr[x])
	{
		int fx=find(arr[x]);
		rank[x]=(rank[x]+rank[arr[x]])%3;
		arr[x]=fx;
	}
	return arr[x];
}
int merge(int x, int y,int type)
{
	int fx=find(x);
	int fy=find(y);
	if(fx==fy)
	{
		if((rank[y]-rank[x]+3)%3!=type)
			return 1;
		else 
			return 0;
	}
	arr[fy]=fx;
	rank[fy]=(rank[x]-rank[y]+type+3)%3;
	return 0;
}
int main()
{
	int n,k;
	int sum,i;
	int d,x,y;
	scanf("%d%d",&n,&k);
	init(n);
	sum=0;
	for(i=1;i<=k;i++)
	{
		scanf("%d%d%d",&d,&x,&y);
		if(x>n||y>n||(x==y&&d==2))
		{
				sum++;
				
		}
		else if(merge(x,y,d-1))
				sum++;
	}
	printf("%d\n",sum);
	return 0;
}