P1196 [NOI2002]銀河英雄伝説・加重互いに素セット

問題の解決策

加重互いに素なセット、3つのアレイを製造：
fa[x]祖先ノードxが示す
sz[x]の集合であるXの大きさを表す
d[x]祖先ノードの内部を設定する距離を表すノード

内部- find(x)：から更新祖先ノードに同時にFA [X]、及び内部ノードの集合を更新
外-コードを参照

---

---

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,k;

int fa[N],sz[N],d[N];//sz[i]表示第i列飞船的总数 d[i]表示第i个飞船到队头的距离
int find(int x){
    if(fa[x]==x)return x;
    int fx=find(fa[x]);
    d[x]+=d[fa[x]];//(队列内部)更新到头的距离
    return fa[x]=fx;
}

char op;
int x,y;
int main(){
    for (int i = 1; i <= N; ++i) {//init
        fa[i]=i;
        sz[i]=1;
        d[i]=0;
    }

    int T;
    scanf("%d",&T);
    for (int cs = 1; cs <= T; ++cs) {
        scanf(" %c%d%d",&op,&x,&y);
        if(op=='M'){//合并 y是新的祖先 题目要求
            x=find(x);
            y=find(y);
            fa[x]=y;//y是新的队列的头
            d[x]+=sz[y];//更新队头x的距离(内部在find函数里面更新)
            sz[y]+=sz[x];//更新队列y的总数
            sz[x]=0;//队列x清零
        }else{
            if(find(x)!=find(y))puts("-1");
            else printf("%d\n", abs(d[x]-d[y])-1);
        }
    }
    return 0;
}