POJ 1988 Cube Stacking

#include<iostream>
#include<string>
using namespace std;
int n;
int p[40000];
int cnt[40000];//记录当前节点到根节点的距离（不是树中的距离，而是在栈中，）
int total[40000];//记录树的总节点个数
int find(int x){
    if(x!=p[x]){
        int t=p[x];
        p[x]=find(p[x]);
        cnt[x]+=cnt[t];
    }
    return p[x];
}
void hb(int x,int y){
    int xx=find(x),yy=find(y);
    p[xx]=yy;
    cnt[xx]+=total[yy];
    total[yy]+=total[xx];
}
int main(){
    cin>>n;
    for(int i=0;i<40000;i++){
        p[i]=i;
        cnt[i]=0;
        total[i]=1;
    }
    while(n--){
        string s;   cin>>s;
        if(s=="M"){
            int x,y;    cin>>x>>y;
            hb(x,y);
        }else{
            int x;  cin>>x;
            find(x);
            cout<<cnt[x]<<"\n";
        }
    }
    return 0;
}