POJ 1988 Cube Stacking 【带权并查集】

#include <iostream>
#include <cstdio>
using namespace std;

const int M= 3e4+10;
int father[M], son[M], dis[M];
//dis表示当前节点到根节点的距离
//son表示当前节点的子树大小(包括该节点本身)
int find(int x){
    if(father[x]==x){
        return x;
    }
    int temp=father[x];  
    father[x]=find(father[x]);
    dis[x]+=dis[temp];  //x到改变前根节点的距离即x到temp的距离加上temp到根节点的距离
    return father[x];
}
void Union(int x, int y){
    int px=find(x);
    int py=find(y);
    if(px!=py){                        
        father[py]=px;   //px为py父亲
        dis[py]=son[px]; //py到x的根节点的距离为合并之前px的子树大小
        son[px]+=son[py];  //更新合并后px的子树大小
    }
}
int main(){
    int p;
    scanf("%d", &p);
    for(int i=1; i<=M; i++){
        father[i]=i;    
        son[i]=1;
    }
    for(int i=1; i<=p;i++){
        char s[2];
        int x, y;
        scanf("%s", s);
        if(s[0]==‘M‘){
            scanf("%d%d", &x, &y);
            Union(x, y);  //以x原来所在列的根为根
        }else{
            scanf("%d", &x);
            printf("%d\n", son[find(x)]-dis[x]-1);//根节点的子树大小减去x到根节点的距离，再减去x本身
        }
    }
    return 0;
}