【洛谷P1196】银河英雄传说

有特殊意义的一道题……

加权并查集，我们增加cnt、deep数组

分别表示i点所在链共有多少个点，以及路径压缩之前点i在链中的深度

每次合并时直接修改cnt，路径压缩的同时更新deep

因为每次查询之前都要find，所以查询节点x的deep也就在找father时更新了

（deep数组初始化为零比较方便）

 #include<cstdio>
 #include<cstring>
 #include<algorithm>

 using namespace std;
 const int n=;
 int father[n+],deep[n+],cnt[n+];
 char c[];

 void pre(){
     for (int i=;i<=n;i++){
         deep[i]=;//**
         father[i]=i;
         cnt[i]=;
     }
 }

 int find(int x){
     if (father[x]!=x){
         int fa=find(father[x]);
         deep[x]+=deep[father[x]];
         father[x]=fa;
     }
     return father[x];
 }

 void merge(int x,int y){
     int r1=find(x);
     int r2=find(y);
     if (r1!=r2){
         father[r1]=r2;
         deep[r1]=cnt[r2];
         cnt[r2]+=cnt[r1];
     }
 }

 int ask(int x,int y){
     int r1=find(x);
     int r2=find(y);
     if (r1!=r2)
         return -;
     return max(,abs(deep[x]-deep[y])-);
 }

 int main(){
     int T,x,y;
     scanf("%d",&T);
     pre();
     for (int i=;i<=T;i++){
         scanf("%s",c);
         scanf("%d %d",&x,&y);
         if (c[]=='M') merge(x,y);
         if (c[]=='C') printf("%d\n",ask(x,y));
     }
     return ;
 }