银河英雄传说

https://www.acwing.com/problem/content/description/240/

难度：简单

时/空限制：2s / 64MB

总通过数：825

总尝试数：1670

有一个划分为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≤500000N≤30000,T≤500000

输入样例：

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

输出样例：

-1
1

思路：并查集边权维护。 

#include<iostream>

using namespace std;

const int N = 30000 + 10;


int p[N];//父节点
int d[N];//到根节点距离；
int siz[N];//当前列大小

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

int main()
{
    int n;
    cin>>n;
    for(int i=1; i<=30000; i++)
    {
        p[i] = i;
        siz[i] = 1;
    }
    
    while(n--)
    {
        char op[2];
        int a,b;
        scanf("%s%d%d",op,&a,&b);
        if(op[0] == 'M')
        {
            int pa = find(a);
            int pb = find(b);
            d[pa] = siz[pb];
            siz[pb] += siz[pa];//合并后数量增加
            p[pa] = pb;
        }
        else
        {
            if(find(a) != find(b)) puts("-1");
            else cout<<max(0, abs(d[a] - d[b]) - 1)<<endl;//可能a=b
        }
    }
    return 0;
}