poj 1182 (带权并查集)

Description

动物王国中有三类动物A,B,C，这三类动物的食物链构成了有趣的环形。A吃B， B吃C，C吃A。
现有N个动物，以1－N编号。每个动物都是A,B,C中的一种，但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述：
第一种说法是"1 X Y"，表示X和Y是同类。
第二种说法是"2 X Y"，表示X吃Y。
此人对N个动物，用上述两种说法，一句接一句地说出K句话，这K句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。
1） 当前的话与前面的某些真的话冲突，就是假话；
2） 当前的话中X或Y比N大，就是假话；
3） 当前的话表示X吃X，就是假话。
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。

Input

第一行是两个整数N和K，以一个空格分隔。
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。
若D=1，则表示X和Y是同类。
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

100 7
1 101 1
2 1 2
2 2 3

2 3 3

1 1 3
2 3 1
1 5 5

Sample Output

3

带权并查集和普通并查集最大的区别在于带权并查集合并的是可以推算关系的点的集合（可以通过集合中的一个已知值推算这个集合中其他元素的值）。而一般并查集合并的意图在于这个元素属于这个集合。带权并查集相当于把“属于”的定义拓展了一下，拓展为有关系的集合。

本题用rank[x]记录x与x的最远的祖先的关系。 这里定义rank[x]=0表示x与x的祖先是同类。rank[x]==1表示x吃x的祖先。rank[x]==2表示x的祖先吃x；这样定义后就与题目中输入数据的D联系起来，(D-1)就可以表示x与y的关系。这样就可以用向量的形式去推关系的公式了。我们用f(x,father[x])表示rank[x]的值；

#include<stdio.h>
int fa[50005]={0};
int rank[50005]={0};
int n;
void initial()
{
    for(int i=1;i<=n;i++)
    {
        fa[i]=i; rank[i]=0;
    }
}
int getfather(int x)
{
    if(x==fa[x])  return x;
    int oldfa = fa[x];
    fa[x]=getfather(fa[x]);
    rank[x]=(rank[x]+rank[oldfa])%3;  //用向量的形式很快就可以看出来
    return fa[x];
}
void unionset(int r,int x,int y)
{
    int fx,fy;
    fx=getfather(x); fy=getfather(y);
    if(fx==fy) return;
    fa[fx]=fy;
    rank[fx]=(rank[y]+r-rank[x]+3)%3;   // 这里同样可以用向量来推公式。另外需要注意的是，这里只更新了fx的rank值，而fx的儿子的rank值都没有更新会不会有问题。其实不碍事，由于我们每次输入一组数据我们都对x和y进行了getfather的操作（x>n || y>n ……)的除外。在执行getfather的操作时，在回溯的过程中就会把fx的儿子的rank值都更新了。
    return ;
}
int istrue(int d,int x,int y)
{
    int fx,fy,r;
    if(x>n || y>n || ((x==y)&&(d==2)) )
        return 0;
    fx=getfather(x); fy=getfather(y);
    if(fx!=fy)  return 1;
    else
    {
        if(rank[x]==((d-1)+rank[y])%3) return 1;  // 这个公式可以用向量来推：如果（ f(x,y) + f(y,father[y]））% 3 == f(x,father[x]) 则是正确的，否则是错的。这个形式可以用向量来表示，就是判断这个向量加法对不对   x--->y + y---> fx(fy) 是否等于  x--->fx(fy)
        else  return 0;
    }
}
int main()
{
    int k,i,x,y,d; int ans=0;
    scanf("%d%d",&n,&k);
    initial();
    for(i=1;i<=k;i++)
    {
        scanf("%d%d%d",&d,&x,&y);
        if( !istrue(d,x,y) )
            ans++;
        else
            unionset(d-1,x,y);
    }
    printf("%d\n",ans);
    return 0;
}