B - 食物链 POJ - 1182 （带权并查集）

动物王国中有三类动物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。

带权并查集：

初学并查集的时候不太理解带权并查集，现在感觉主要是理解一下偏移量 和 路径压缩过程中偏移量的变化，就应该很好理解带权并查集了。

首先考虑路径压缩。

int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
这样可以做到每次find后，x的祖先要么是它本身，要么是它的父亲节点，树的深度不超过2。

 

令val[u]表示节点u到父亲节点fu的偏移量，val[v]表示节点v到父亲节点fv的偏移量。

加入新的边 (u,v,w)时，若令fa[fu]=fv，则有val[fu] = val[v]+w-val[u]

上图中u到v的偏移量就为 val[u]+val[v]+w-val[u] - val[v] = w，即u到新的根节点的偏移量减去v到新的根节点的偏移量，显然与新加入的边效果是一样的。

假设此后加入新的边 (u,v1,w1)

在执行find(u)操作时，进行路径压缩，偏移量的变化其实就是把原来路径上的总偏移量都加上，形成新的偏移量

比如下图val[u]+val[v]+w-val[u]=val[v]+w，路径压缩后u到fv的偏移量就是val[v]+w。

路径压缩后连边又和起始一样了。

int find(int x){
    if(fa[x]!=x){
        int tt=fa[x];
        fa[x]=find(fa[x]);
        val[x]=val[x]+val[tt];
    }
    return fa[x];
}
本题由于三种动物形成了一个环，那么偏移量应该始终在0-2之间，即对3取模。

（吐槽：HDU有些题不提示多组数据；POJ则有很多单组数据的题，输入后可能有些奇奇怪怪的东西，用判断EOF的方式读入就会出错） 

代码：

//Full of love and hope for life

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <queue>
#include <vector>
#define inf 0x3f3f3f3f
//https://paste.ubuntu.com/
//https://www.cnblogs.com/zzrturist/    //博客园
//https://blog.csdn.net/qq_44134712     //csdn

using namespace std;
const int N=5e3+10;
typedef long long ll;
const int mod=1e9+7;

int disv[50010];
int f[50010];

int getx(int x){//带权并查集
    if(x==f[x]){
        return x;
    }
    int root=getx(f[x]);
    disv[x]=(disv[x]+disv[f[x]])%3;
    return f[x]=root;
}

signed main(){//此题必须用scanf和printf，不然会tle
    int a,b,ans;
    int u,v,w;
    scanf("%d%d",&a,&b);
    ans=0;
    for(int i=1;i<=a;i++){
        f[i]=i;
    }
    for(int i=1;i<=b;i++){
        scanf("%d%d%d",&w,&u,&v);
        w--;
        if(u>a||v>a||u<1||v<1){
            ans++;
            continue;
        }
        int dx=getx(u);
        int dy=getx(v);
        if(dx==dy){//权值相等
            if(((disv[u]-disv[v]+3)%3)!=w){
                ans++;
            }
        }
        else//权值不相等
        {
            f[dx]=dy;
            disv[dx]=(w-disv[u]+disv[v]+3)%3;
        }
    }
    printf("%d",ans);
    return 0;
}

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