【洛谷P1196】银河英雄传说

有特殊意义的一道题……

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

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

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

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

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

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

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

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

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 -1;
    return max(0,abs(deep[x]-deep[y])-1);
}

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