【并查集】ACW238. 银河英雄传说

#include <iostream>
#include <cstdlib>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>
#include <functional>
#include <unordered_set>
#include <unordered_map>
#include <set>
#include <bitset>
#include <cstdio>


using namespace std;

const int N = 1e6 + 100;

// 维护一个根数组，一个所在位置序号数组
// 一个如果i是根节点这个集合的元素数量
int p[N], nums[N], state[N];

inline void init() {
    for (int i = 1; i < 3e4 + 10; i++) {
        p[i] = i;
        nums[i] = 1;
    }
}

int find(int x) {
    if (p[x] != x) {
        int root = find(p[x]);
        // 向上递归更新此时x前面的元素个数
        // 使用递归的话，每个father节点也会更新
        state[x] += state[p[x]];
        p[x] = root;
    }
    return p[x];
}

void merge(int x, int y) {
    int fa = find(x), fb = find(y);
    if (fa != fb) {
        // 此时将x所在的列接在y所在的列的后面
        state[fa] = nums[fb];
        // 更新y所在列的元素个数
        nums[fb] += nums[fa];
        p[fa] = fb;
    }
}

int main() {
    ios::sync_with_stdio(false);
    int T;
    cin >> T;
    init();
    while (T--) {
        char op;
        int x, y;
        cin >> op >> x >> y;
        if (op == 'M') merge(x, y);
        else {
            if (find(x) != find(y)) {
                cout << -1 << endl;
            } else {
                cout << max(0, abs(state[x] - state[y]) - 1);
            }
        }
    }
    return 0;
}