洛谷 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+],cnt[N+],siz[N+];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',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=;i<=N;i++) f[i]=i,siz[i]=;
     for(rg int i=;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]=;
         }
         else printf("%d\n",find(x)==find(y)?abs(cnt[x]-cnt[y])-:-);
     }
     return ;
 }