并查集——银河英雄传说（）

传送门：238. 银河英雄传说 - AcWing题库

思路：
使用并查集可以传递关系的性质，维护一个cnt[i]数组，该数组用于记录以i为跟并查集树下的战舰的数量，用一个d[i]数组表示在i前面的战舰的数量。

d数组和cnt数组的维护可以在get函数和merge函数时实现。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
typedef long long ll;
const int N=3e4+10;
int d[N],f[N],cnt[N];
int get(int x)
{
    if(x==f[x]) return x;
    int u=get(f[x]);
    d[x]+=d[f[x]];
    return f[x]=u;
}
void merge(int x,int y)
{
    x=get(x),y=get(y); //找到x，y分别所在队列的首节点战舰
    f[x]=y;
    d[x]=cnt[y];  //原本一个首节点前面的战舰数量是零，但是现在接在了y后面，所以要加上cnt【y】，x后面的节点的d值会在下一次有关查找时更新。
    cnt[y]+=cnt[x];    //因为是将x所在的队列接到y后面所以是y加上x
}
int main()
{
    int t;
    for(int i=1;i<=N;i++)
    {
        f[i]=i;
        cnt[i]=1;//每艘战舰自身的便算一个
        //d[i]=1;  //这里不能初始化为1是因为一开始i前面的战舰的数量为零。
    }
    scanf("%d",&t);
    while(t--)
    {
        char a[2];
        int b,c;
        scanf("%s%d%d",a,&b,&c);
      int   x=get(b),y=get(c);
        if(a[0]=='C')
        {
            if(x==y)
            {
                printf("%d\n",max(0,abs(d[b]-d[c])-1));
            }else
            puts("-1");
        }else
        if(x!=y)
        merge(x,y);
    }
    return 0;
}