（带权并查集）POJ1182 食物链

POJ1182 食物链

思路：

因为题目给的相对关系，所以可以用带权并查集做。
权值：0表示x和y是同类，1表示x吃y，2表示x被y吃。
那么在路径压缩的时候，我们得到w[a->c]=(w[a->b]+w[b->c])%3。在区间合并的时候w[t1]=(-w[x]+w[y]+w-1)%3。在判断关系的时候比较w-1和（w[x]-w[y]+3)%3，+3是为了避免负数，w-1是因为本题给的权值为1，2。

代码：

#define ll long long
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<ctype.h>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
const int N=5e4+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
int fa[N],sum[N]={0},ans=0;
int find(int x)
{
	if(x!=fa[x])
	{
		int t=fa[x];
		fa[x]=find(fa[x]);
		sum[x]=(sum[x]+sum[t])%3;
	}
	return fa[x];
}
void ask(int u,int v,int w)
{
	int t1=find(u),t2=find(v);
	if(t1==t2 && w-1!=(sum[u]-sum[v]+3)%3)
		ans++;
	else if(t1!=t2)
	{
		fa[t1]=t2;
		sum[t1]=(-sum[u]+sum[v]+w-1)%3;
	}
	return;
}
int main()
{
	int n,m,i;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
		fa[i]=i;
	while(m--)
	{
		int x,y,re;
		scanf("%d%d%d",&re,&x,&y);
		if(x>n || y>n || (x==y && re==2))
			ans++;
		else
			ask(x,y,re);
	}
	printf("%d\n",ans);
	return 0;
}