CH4101银河英雄传说(带权并查集)

题目链接：https://ac.nowcoder.com/acm/contest/1031/B

 

题目大意：有n列战舰，两种命令，第一种命令是把i号战舰所在列接在j号战舰所在列后面，第二种命令是若第i号战舰和第j号战舰不在同一列输出-1，否则输出它们之间的战舰数量

 

题目思路：设置一个d[x]表示x前面还有几个人，size[x]表示x这一列有多少人(x是队首)，然后路径压缩的时候，每个人压缩得加上父亲前面还有几个人，这样就能知道每个人离队首的距离，至于合并就是被接到后面去的队首的d[x]变成前面那队队伍人数，然后前面那个队的size+新来的队伍的size即可。

 

以下是代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
const int MAXN = 1e5+5;
const int MOD = 1e9+7;
char s[5];
int pre[MAXN],d[MAXN],siz[MAXN];
int Find(int x){
    if(x==pre[x])return x;
    int rt=Find(pre[x]);
    d[x]+=d[pre[x]];
    return pre[x]=rt;
}
int main(){
    int t,x,y;
    scanf("%d",&t);
    rep(i,1,3e4){
        pre[i]=i;
        d[i]=0,siz[i]=1;
    }
    while(t--){
        scanf("%s%d%d",s,&x,&y);
        if(s[0]=='M'){
            int xx=Find(x),yy=Find(y);
            pre[xx]=yy;
            d[xx]=siz[yy];
            siz[yy]+=siz[x];
        }
        else{
            int xx=Find(x),yy=Find(y);
            if(xx!=yy)cout<<-1<<endl;
            else printf("%d\n",abs(d[y]-d[x])-1);
        }
    }
    return 0;
}