P1196 [NOI2002] 银河英雄传说

链接：https://www.luogu.com.cn/problem/P1196

调了挺久的，一些细节没注意到。
细节在注释里都标注了

#include<bits/stdc++.h>
using namespace std;
const int maxn=50000;
int f[maxn];
int dis[maxn];//战舰在行中的位置
int num[maxn];//行中有几个战舰
void init(){
    for(int i=0;i<maxn;i++){
        f[i]=i;
        dis[i]=0;
        num[i]=1;
    }
}

int find(int x){
    if(f[x]==x){
        return x;
    }
    int fa=f[x];
    f[x]=find(f[x]);
    dis[x]+=dis[fa];
    return f[x];
    //所有该行战舰距离改变值等于头战舰的距离改变值
    //一行战舰合并到另一行之后，头战舰已经改变了，所以不用担心重复加多次的问题
}
void merge(int y,int x){
    int a=find(x);
    int b=find(y);
    f[b]=a;
    dis[b]+=num[a];//更新头战舰所处位置
    num[a]+=num[b];//更新行长度
    num[b]=0;
}
int main(){
    init();
    int t;
    cin>>t;
    char c;
    int aa,bb;
    for(int i=0;i<t;i++){
        cin>>c>>aa>>bb;
        if(c=='M'){
            merge(aa,bb);
        }
        if(c=='C'){
            int a=find(aa);
            //这里必须是find(aa)，如果是find(f[aa])会导致叶子结点距离不被更新
            int b=find(bb);
            if(a!=b){
                cout<<-1<<endl;
            }else{
                cout<<abs(dis[bb]-dis[aa])-1<<endl;//算的是中间隔了几个，要有-1
            }
        }
    }
    //system("pause");
    return 0;
}