poj1182食物链（种类,带权并查集)

题目大意：有三类不同动物，给出给K句话，描述两个动物间是同类还是捕食关系，若与前面的真话冲突，则判定为假话，输出假话的数量

思路分析：可用并查集将出现已知关系的动物合并在一起，各个动物到根结点距离对3的不同余数表示不同的种类，如对3的余数为0，1，2分别表示不同种的动物。当要将两物种合并在一起时，可通过修改其中一个跟结点的权值使这两种动物满足关系

#include<iostream>
#include<stdio.h>
using namespace std;
int p[100000],A[100000],n;
int f(int x){
if(x==p[x]) return x; int F=f(p[x]); //路径压缩，同时更新该结点的权值，这条语句不可以和下面的更改顺序，要先更新其原父节点的值（初始化时为0）才能更新其值。因为下面每次更新时只更新了一个集合根结点的值，但集合中其他元素的值未变，所以查询一个点的权值，可从父节点一步步更新
A[x]=(A[p[x]]+A[x])%3;
p[x]=F;
return F;
}
int u(int x,int a,int b){
int f1=f(a),f2=f(b);
if(a>n||b>n) return 0;
if(x==1){
if(f1==f2&&A[a]!=A[b]) return 0;
if(f1!=f2){ //当a和b不在同一个集合时，合并集合，同时修改a根结点f1的值，使合并后满足a和b同类
A[f1]=(A[b]-A[a]+3)%3;
p[f1]=f2;
}
}
else{
if(f1==f2){
if(A[a]==0&&A[b]!=1) return 0;
if(A[a]==1&&A[b]!=2) return 0;
if(A[a]==2&&A[b]!=0) return 0;
}
else{ //修改f1，使修改后满足a吃b，
A[f1]=(A[b]-A[a]+2)%3;
p[f1]=f2;
}
}
return 1;
}
int main(){
int a,b,c,m,i,sum=0;
scanf("%d%d",&n,&m);
for(i=0;i<=60000;i++){ //初始化并查集，同时使所有的点初始为0
A[i]=0;
p[i]=i;
}
for(i=1;i<=m;i++){
scanf("%d%d%d",&a,&b,&c);
if(u(a,b,c)==0) sum++;
}
printf("%d\n",sum);
return 0;
}