[luoguP1196] 银河英雄传说（并查集）

传送门

记录 up[x] 表示 x 上方有多少个

　　 all[x] 表示当前连通的有多少个

find 的时候 和 合并的时候 更新一下即可

——代码

#include <cstdio>
#include <iostream>
#define N 30001
#define abs(x) ((x) < 0 ? -(x) : (x))

int T;
int f[N], up[N], all[N];

inline int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -1;
    for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
    return x * f;
}

inline int find(int x)
{
    if(x ^ f[x])
    {
        int fx = f[x];
        f[x] = find(f[x]);
        up[x] += up[fx];
    }
    return f[x];
}

int main()
{
    int i, x, y, fx, fy;
    char s[1];
    T = read();
    for(i = 1; i < N; i++) f[i] = i, all[i] = 1;
    while(T--)
    {
        scanf("%s", s);
        x = read();
        y = read();
        fx = find(x);
        fy = find(y);
        if(s[0] == 'M')
        {
            f[fx] = fy;
            up[fx] += all[fy];
            all[fy] += all[fx];
        }
        else fx ^ fy ? puts("-1") : printf("%d
", abs(up[x] - up[y]) - 1);
    }
    return 0;
}