238. 银河英雄传说(并查集，扩展域)

​​​​​​238. 银河英雄传说 - AcWing题库

分析：

由于要求i，j战舰之间的距离，我们就应该想到拓展域

拓展域要实现：

• 能够计算第i艘战舰到第j艘战舰距离

我们可以用一个拓展域来存储战舰到根节点的距离，然后还需要考虑怎么合并拓展域

根据食物链的思想，可以维护，注意根结点到根结点距离为0，避免重复计算

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

还需要维护一列上（一个集合内）有几艘战舰，再开一个拓展域保存

那么合并一个集合的操作如下

int px = find(x), py = find(y);
p[px] = py;
d[px] = siz[py];    //px放在最末尾，离根结点siz距离
siz[py] += siz[px];//增加siz

扩展域：

• d数组           d[i]存第i艘战舰到根结点距离
• siz数组         siz[i]存i所在列的战舰数量

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int p[N];
int siz[N];
int d[N];
int find(int x)
{
    if (x != p[x])
    {
        int u = find(p[x]);
        d[x] += d[p[x]];
        p[x] = u;
    }
    return p[x];
}

int main()
{
    int t;
    cin >> t;
    for (int i = 1;i < N;i++)
        p[i] = i, siz[i] = 1;
    while (t--)
    {
        int x, y;
        char k;
        cin >> k >> x >> y;
        if (k == 'M')
        {
            int px = find(x), py = find(y);
            p[px] = py;
            d[px] = siz[py];
            siz[py] += siz[px];
        }
        else
        {
            int fa = find(x), fb = find(y);
            if (fa == fb)//由于可能是-1，所以max(0,..)
                cout << max(0,abs(d[x] - d[y])-1) << "\n";
            else cout << -1 << "\n";
        }
    }
    return 0;
}