POJ 1182 带权并查集

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 89563		Accepted: 26905

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

Source

Noi 01

这题我们怎么看呢 可以通过列关系得到并查集查找的时候的关系式

然后用某大牛的向量方法 得出答案 这也太巧妙了吧

由上面可知：
    x->y 偏移量0时 x和y同类
    
    x->y 偏移量1时 x被y吃
    
    x->y 偏移量2时 x吃y
    
    有了这个假设，我们就可以在并查集中完成任意两个元素之间的关系转换了。
    
    不妨继续假设，x的当前集合根节点rootx，y的当前集合根节点rooty，x->y的偏移值为d-1（题中给出的询问已知条件）
    
    (1)如果rootx和rooty不相同，那么我们把rooty合并到rootx上，并且更新relation关系域的值（注意：p[i].relation表示i的根结点到i的偏移量！！！！（向量方向性一定不能搞错））
    
        此时 rootx->rooty = rootx->x + x->y + y->rooty，这一步就是大牛独创的向量思维模式
    
        上式进一步转化为：rootx->rooty = (relation[x]+d-1+3-relation[y])%3 = relation[rooty]，（模3是保证偏移量取值始终在[0,2]间）
    
    (2)如果rootx和rooty相同(即x和y在已经在一个集合中，不需要合并操作了，根结点相同)，那么我们就验证x->y之间的偏移量是否与题中给出的d-1一致
    
        此时 x->y = x->rootx + rootx->y
    
        上式进一步转化为：x->y = (3-relation[x]+relation[y])%3，
        若一致则为真，否则为假。
  

  #include <cstdio>
    #include <iostream>
    #include <cstring>
    #include <algorithm>
    #include <cmath>
    using namespace std;
    const int MAX_N = 50024;
    int father[MAX_N],rank_[MAX_N];
    int Find(int x){
        if(father[x]==x){
            return x;
        }
        int y = father[x];
        father[x] = Find(y);
        rank_[x] = (rank_[x]+rank_[y])%3;
        return father[x];
    }
    int main(){
        int n,k,ans = 0;
        scanf("%d%d",&n,&k);
        for(int i = 1;i<=n;++i){
            father[i] = i;
            rank_[i] = 0;
        }
        while(k--){
            int q,a,b;
            scanf("%d%d%d",&q,&a,&b);
            if(a>n||b>n){
                ans++;
                continue;
            }
            if(q==2&&a==b){
                ans++;
                continue;
            }
            int fa = Find(a);
            int fb = Find(b);
            if(fa!=fb){
                father[fb] = fa;
                rank_[fb] = (3+rank_[a]+q-1-rank_[b])%3;
            }
            else {
                if(q==1){
                    if(rank_[a]!=rank_[b]) {
                        ans++;
                        continue;
                    }
                }
                else if(q==2){
                    if((3-rank_[a]+rank_[b])%3!=q-1){
                        ans++;
                        continue;
                    }
                }
            }
        }
        printf("%d\n",ans);
        return 0;
    }