POJ 1182 食物链

带权并查集，点权表示自己与父亲的关系，0同类，1被父亲吃，2吃父亲。

这样记可以很方便地转移关系，可以证明关系转移的表达式:自己和爷爷的关系=（自己和父亲的关系+父亲和爷爷的关系）mod 3

路径压缩，判断假话，都要用到这个转移式。

#include<cstdio>
#define MAXN 50005
int n;
struct animal
{
    int fa, r;
}ani[MAXN];
void init()
{
    for(int i = 1; i <= n; i++)
        ani[i]=(animal){i,0};
}
int find(int x)
{
    if(ani[x].fa==x)return x;
    find(ani[x].fa);
    ani[x].r=(ani[x].r+ani[ani[x].fa].r)%3;
    ani[x].fa=ani[ani[x].fa].fa;
    return ani[x].fa;
}
void uni(int fx, int x, int y, int d)
{
    ani[fx].r=(d+3-ani[x].r)%3;
    ani[fx].fa=y;
}
int main()
{
    int ans=0,d,x,y,k;
    scanf("%d%d",&n,&k);
    init();
    for(int i = 1; i <= k; i++)
    {
        scanf("%d%d%d",&d,&x,&y);
        if(x>n||y>n){ans++;continue;}
        int fx=find(x), fy=find(y);
        if(d==1)
        {
            if(fx==fy && ani[x].r!=ani[y].r)
                ans++;
            else if(fx!=fy)
                uni(fx,x,y,0);
        }
        else
        {
            if(fx==fy && ani[x].r!=(d+ani[y].r)%3)
                ans++;
            else if(fx!=fy)
                uni(fx,x,y,2);
        }
    }
    printf("%d\n",ans);
    return 0;
}