poj 1182 食物链

并查集进阶，除了用父亲数组保存父结点外，再用一个数组表示子结点与父结点之间的关系，查找与合并时都涉及到关系数组的修改。

#include <stdio.h>
#include <string.h>
int rank[50010],fa[50010];
int find(int n)
{
    int t = fa[n];
    if(fa[n] != n)
        fa[n] = find(fa[n]);
    rank[n] = (rank[n]+rank[t])%3;
    return fa[n];
}
void uni(int x,int y,int h)
{
    int a = find(x);
    int b = find(y);
    fa[a] = b;
    rank[a] = (rank[y]-rank[x]+3+h)%3;
}
int main()
{
    int n,k,i,f,a,b,cnt=0;
    scanf("%d%d",&n,&k);
    for(i = 0; i < n; i++)
        fa[i] = i;
    while(k--)
    {
        scanf("%d%d%d",&f,&a,&b);
        if(a>n || b>n)
        {cnt++; continue;}
        if(f==1)
        {
            if(find(a)==find(b))
            {
                if(rank[a]!=rank[b])
                    cnt++;
            }
            else uni(a,b,0);
        }
        else
        {
            if(find(a)==find(b))
            {
                if(rank[a]!=(rank[b]+1)%3)
                    cnt++;
            }
            else uni(a,b,1);
        }
    }
    printf("%d",cnt);
    return 0;
}