洛谷 1196 [NOI2002]银河英雄传说【模板】带权并查集

【题解】

　　经典的带权并查集题目。

　　设cnt[i]表示i前面的点的数量，siz[i]表示第i个点(这个点是代表元)所处的联通块的大小；合并的时候更新siz、旧的代表元的cnt，路径压缩的时候维护cnt即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define rg register
#define N 30000
using namespace std;
int n,m,f[N+10],cnt[N+10],siz[N+10];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
} 
int find(int x){
    if(f[x]==x) return x;
    else{
        int fa=find(f[x]);
        cnt[x]+=cnt[f[x]];
        return f[x]=fa;
    }
} 
int main(){
    n=read();
    for(rg int i=1;i<=N;i++) f[i]=i,siz[i]=1;
    for(rg int i=1;i<=n;i++){
        char c=getchar(); while(c!='M'&&c!='C') c=getchar();
        int x=read(),y=read();
        if(c=='M'){
            f[x=find(x)]=(y=find(y));
            cnt[x]+=siz[y];
            siz[y]+=siz[x]; siz[x]=0;
        }
        else printf("%d
",find(x)==find(y)?abs(cnt[x]-cnt[y])-1:-1);
    }
    return 0;
}