银河英雄传说 （并查集保存到根节点距离）

有一个划分为 N 列的星际战场，各列依次编号为 1,2,…,N。有 N艘战舰，也依次编号为1,2,…,N，其中第 i 号战舰处于第 i列。有 T条指令，每条指令格式为以下两种之一：

1. M i j，表示让第 i号战舰所在列的全部战舰保持原有顺序，接在第 j号战舰所在列的尾部。
2. C i j，表示询问第 i号战舰与第 j号战舰当前是否处于同一列中，如果在同一列中，它们之间间隔了多少艘战舰。现在需要你编写一个程序，处理一系列的指令。

 输入格式

第一行包含整数 T，表示共有 T条指令。接下来 T行，每行一个指令，指令有两种形式：M i j 或 C i j。其中 M和 C 为大写字母表示指令类型，i 和 j为整数，表示指令涉及的战舰编号。

输出格式

你的程序应当依次对输入的每一条指令进行分析和处理：

如果是 M i j 形式，则表示舰队排列发生了变化，你的程序要注意到这一点，但是不要输出任何信息；

如果是 C i j 形式，你的程序要输出一行，仅包含一个整数，表示在同一列上，第 i号战舰与第 j 号战舰之间布置的战舰数目，如果第 i 号战舰与第 j 号战舰当前不在同一列上，则输出 −1。

数据范围

N≤30000,T≤500000

输入样例：

4
M 2 3
C 1 2
M 2 4
C 4 2

输出样例：

-1
1

思路：

由于要查询 i 到 j 战舰之间的战舰数，所有要保存到根节点距离。

代码：

#include <bits/stdc++.h>
using namespace std;

const int N = 30010;
int n, m;
int p[N],sz[N],d[N];//p是并查集，sz是根代表的集合中元素个数，d代表每个点到跟的距离

int find(int x) {
    if (p[x] != x) {
        int root = find(p[x]);
        d[x] += d[p[x]];
        p[x] = root;
    }
    return p[x];
}

int main()
{
    scanf("%d", &m);
    for (int i = 1; i < N; i++) {
        p[i] = i;
        sz[i] = 1;
    }

    while (m--) {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);
        if (op[0] == 'M') {
            int pa = find(a), pb = find(b);
            if (pa != pb) {
                d[pa] = sz[pb];
                sz[pb] += sz[pa];
                p[pa] = pb;
            }
        }
        else {
            int pa = find(a), pb = find(b);
            if (pa != pb)cout << "-1" << endl;
            else cout << max(0, abs(d[a] - d[b]) - 1) << endl;
        }
    }
    return 0;
}