食物链(带权并查集)

食物链
并查集维护的每个点到根节点的距离
余1表示可以吃掉根节点
余2表示可以被根节点吃掉
余0表示与根节点同类

//食物链(带权并查集)
#include <bits/stdc++.h>
using namespace std;
const int N = 50010;

int n,m;
int p[N],d[N];

int find(int x)
{
    if(p[x] != x)
    {
        int t = find(p[x]);
        d[x] += d[p[x]];
        p[x] = t;
    }
    return p[x];
}
int main()
{
    int n,m;
    cin >> n >> m;
    for(int i = 1;i <= n;i++)p[i] = i;

    int ans = 0;
    while(m--)
    {
        int t,x,y;
        cin >> t >> x >> y;
        if(x > n || y > n){ans++;continue;}
        int a = find(x),b = find(y);
        if(t == 1)
        {
            if(a == b && (d[x] - d[y]) % 3)ans++;//在同一个集合但是不同种
            else if(a != b)
            {
                p[a] = b;
                d[a] = d[y] - d[x];
            }
        }
        else 
        {
            if(a == b && (d[x] - d[y] - 1) % 3)ans++;//同一个集合但是不是y吃x的类型
            else if(a != b)
            {
                p[a] = b;
                d[a] = d[y] + 1 - d[x];
            }
        }
    }
    cout << ans;
    return 0;
}