维护带权路径长度并压缩的并查集

https://www.acwing.com/problem/content/242/

#include<iostream>
using namespace std;

const int N=100010;
int n,m;
int p[N],d[N];


int find(int x){
    if(p[x]!=x){
        int u=p[x];//save father
        p[x]=find(p[x]); //press
        d[x]+=d[u]; //update path
    }
    return p[x];
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)   p[i]=i;
    int res=0;
    while(m--){
        int t,x,y;
        cin>>t>>x>>y;
        if(x>n || y>n) res++;//越界，错误情况1
        else{
            int px=find(x),py=find(y); //find会执行路径压缩，指向根代表
            if(t==1){   //声明同类
                if(px==py && (d[x]-d[y])%3) res++;//同根且不同类型，错误情况2 
                else if(px!=py){//不同根，转化为同根。
                    p[px]=py;
                    d[px]=d[y]-d[x];
                }
            }else{     //声明x吃y
                if(px==py && (d[x]-d[y]-1)%3) res++; //同根且类型相同或者y吃x，错误情况3
                else if(px!=py){//不同根转换为同根，注意合并时距离关系
                    p[px]=py;
                    d[px]=d[y]+1-d[x]; //x%3 一定要比 y%3 大1才会有X吃Y
                }
            }
        }
    }
    cout<<res;
    
    return 0;
}