【带权并查集】银河英雄传说

洛谷P1196 [NOI2002]银河英雄传说

一道喜闻乐见的带权并查集qwq

出了以下锅

• 没在主程序中写上预处理//撞墙吧
• int k = f[x]   dis[x] += dis[k]
• 写成 dis[x] += dis[f[x]]
• 意识到f[x]在变以后把这一句写到了f[x] = find(f[x]) 前面QAQ
• 可是这个dis[x] 是由x的祖先一步步推回来的啊QAQ 

贴代码QAQ

#include<cmath>
#include<cstdio>
#include<iostream>
using namespace std;
const int sz = 30030;
int f[sz], num[sz], dis[sz];
int T, ans = 0;
char opt[2];
void init() {
    for(int i = 1; i <= 30000; i++) {
        f[i] = i;
        num[i] = 1;
        dis[i] = 0;
    }
}
int find(int x) {
    if(x != f[x]) {
        int k = f[x];
        f[x] = find(f[x]);
        dis[x] += dis[k];
        num[x] = num[f[x]];
    }
    return f[x];
}
int merge(int u, int v) {
    int r1 = find(u), r2 = find(v);
    f[r1] = r2;
    dis[r1] = dis[r2] + num[r2];
    num[r2] += num[r1];
    num[r1] = num[r2];
}
int main() {
//    freopen("qaq.out","w",stdout);
    scanf("%d",&T);
    init();
    while(T--) {
        int i, j;
        scanf("%s",opt+1);
        if(opt[1]=='M') {
            scanf("%d%d",&i, &j);
            merge(i, j);
        }
        else {
            scanf("%d%d",&i, &j);
            if(find(i) != find(j)) {
                printf("-1\n");
                continue;
            }
            else {
                ans = abs(dis[i] - dis[j]) - 1;
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}