【NOIP2002】银河英雄传说

本题在洛谷上的链接：https://www.luogu.org/problemnew/show/P1196

---

 

又是一道带权并查集的题，没太多难度，就是吧。。。

为了提高代码能力，换了一种更松散的代码风格，导致之前的读入优化有一行很长，所以就去抄了某位大佬的模板，结果被带跑了。。。

读入字符时傻了一样“跳”着读，结果全TLE了。。。

说正事，显然我们需要维护每个结点到其祖先的距离，而这个距离恰好是该结点之前的结点个数，因此我们还需要维护每个并查集的结点个数，注意，只需要维护祖先的就可以，其他结点可以找到祖先，然后查询祖先的即可。

然后就是查询时要加绝对值，然后还要减1，这题坑基本没有，还算友好。

#include <cstdio>
#include <cmath>

inline int get_num() {
    int num = 0;
    char c = getchar();
    while (c < '0' || c > '9') c = getchar();
    while (c >= '0' && c <= '9')
        num = num * 10 + c - '0', c = getchar();
    return num;
}

const int maxn = 3e4 + 5;

int fa[maxn], dist[maxn], size[maxn];

int dj_find(int i) {
    if (i == fa[i]) return i;
    int old = fa[i];
    fa[i] = dj_find(old);
    dist[i] += dist[old];
    return fa[i];
}

inline void dj_merge(int a, int b) {
    a = dj_find(a), b = dj_find(b);
    if (a != b) {
        fa[a] = b;
        dist[a] += size[b];
        size[b] += size[a];
    }
}

int main() {
    for (int i = 1; i <= 30000; ++i)
        fa[i] = i, dist[i] = 0, size[i] = 1;
    int t, a, b;
    char c;
    t = get_num();
    for (int i = 1; i <= t; ++i) {
        c = 0; //此处史无前例的坑！
        while (c != 'M' && c != 'C')
            c = getchar();
        a = get_num(), b = get_num();
        if (c == 'M') dj_merge(a, b);
        else {
            if (dj_find(a) != dj_find(b)) printf("-1\n");
            else printf("%d\n", abs(dist[a] - dist[b]) - 1);
        }
    }
    return 0;
}