AcWing 238. 银河英雄传说

思路

其实就是个简单的并查集，只不过这里需要知道两个战舰的长度，需要len数组存一下某个根节点队列的长度，例如：len[2]就是以2为根节点的队列长度，那么两个集合合并的时候，将y集合接到x集合尾部，d[i]表示i节点到根节点的距离的话，d[y的根节点]=len[x]，在更新len[x]+=len[y]即可

代码

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=30010;
int p[N];
int d[N];
int len[N];//记录某一个根节点队列长度
int n;
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(){
    int t;
    cin >> t;
    for (int i = 1; i < N; i ++ )
    {
        p[i] = i;
        len[i] = 1;
    }
    while(t--){
        char k;int a,b;
        cin >> k >> a >> b;
        int pa=find(a);int pb=find(b);
        if(k=='M'){
            p[pa]=pb;
            d[pa]+=len[pb];
            len[pb]+=len[pa];//更新一下杯接的根节点长度
        }else{
            if(pa!=pb){
                cout << -1 << "\n";
            }else{
                 cout << max(0, abs(d[a] - d[b]) - 1) << "\n";//特判一下相邻的和同一个战舰
            }
        }
    }
}