食物链——带权并查集 / 带扩展域并查集

动物王国中有三类动物 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 句话有的是真的，有的是假的。
当一句话满足下列三条之一时，这句话就是假话，否则就是真话。
当前的话与前面的某些真的话冲突，就是假话；
当前的话中 X 或 Y 比 N 大，就是假话；
当前的话表示 X 吃 X，就是假话。
你的任务是根据给定的 N 和 K 句话，输出假话的总数。

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

输出格式
只有一个整数，表示假话的数目。

数据范围
1≤N≤50000,0≤K≤100000

输入样例：
100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例：
3

一、带权并查集

#include <bits/stdc++.h>
using namespace std;
#define int long long 
const int N=2e6+10;
int p[N],d[N];
int find(int x)
{
    if (x!=p[x]) 
    {
        int root=find(p[x]);
        d[x] +=d[p[x]];
        p[x]=root;
    }
    return p[x];
}
signed main()
{
    int n,k;
    cin>>n>>k;
    for (int i=0;i<=n;i++) p[i]=i;
    int ans=0;
    while (k--)
    {
        int c,a,b;
        cin>>c>>a>>b;
        if (a>n||b>n) ans++;
        else if (c==1)
        {
            int x=find(a),y=find(b);
            if (x==y&&(d[a]-d[b])%3) ans++;
            if (x!=y) 
            {
                p[x]=y;
                d[x]=d[b]-d[a];
            }
        }
        else if (c==2)
            {
                int x=find(a),y=find(b);
                if (x==y&&(d[a]-d[b]-1)%3) ans++;
                if (x!=y)
                {
                    p[x]=y;
                    d[x]=d[b]-d[a]+1;
                }
            }
    }
    cout<<ans;
    return 0;
}

二、带扩展域并查集

#include <bits/stdc++.h>
using namespace std;
#define int long long 
const int N=5e4+10,M=3*N;
int p[M];
int find(int x)
{
    if (x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
signed main()
{
    int n,k;
    cin>>n>>k;
    for (int i=0;i<M;i++) p[i]=i;
    int ans=0;
    while (k--)
    {
        int c,a,b;
        cin>>c>>a>>b;
        if (a>n||b>n)
        {
            ans++;
            continue;
        }
    
        if (a==b)
        {
           if (c==2) ans++;
           continue;
        }
        if (c==1)
        {
           if (find(a)==find(b+N)||find(a+N)==find(b))    //若 a吃b 或 b吃a 成立，则a与b是同类不成立
           {
               ans++;
               continue;
           }
           p[find(a)]=find(b);            //为真话时，要记得维护 a与b 的关系
           p[find(a+N)]=find(b+N);
           p[find(a+2*N)]=find(b+2*N);
        }
        else 
        {
           if (find(a+N)==find(b)||find(a)==find(b))     //若 b吃a 或 a与b是同类 成立，则a吃b不成立
           {
               ans++;
               continue;
           }
           p[find(a)]=find(b+N);          //为真话时，要记得维护 a与b 的关系
           p[find(a+N)]=find(b+2*N);
           p[find(a+2*N)]=find(b);
        }
    }
    cout<<ans;
    return 0;
}