POJ 1988 Cube Stacking（并查集）

Description
有n个从1到n(1<=n<=30000)编号的箱子,将每个箱子当做一个栈,对这些箱子进行p次操作 (1<=p<=100000)，操作分两种：
1.M x y:将编号为x的箱子所在的栈放在编号为y的箱子所在栈的栈顶
2.C x:询问编号为x的所表示的栈中在x号箱子下面的箱子数目
Input
第一行为一个整数p表示操作数，之后p行每行为一次操作
Output
对于每次查询，输出查询结果
Solution
并查集，每个栈看做一个集合，用这个栈的栈底元素代表这个集合的父亲元素，用dis[i]表示i下面箱子数量，num[i]表示i所在栈的箱子数量，具体的find和unite函数见代码
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 33333
int fa[maxn],dis[maxn],num[maxn];
void init()
{
    for(int i=1;i<maxn;i++)
        fa[i]=i,dis[i]=0,num[i]=1;
}
int find(int x)
{
    if(x==fa[x])return fa[x];
    int f=fa[x];
    fa[x]=find(fa[x]);
    dis[x]+=dis[f];
    return fa[x];
}
void unite(int x,int y)
{
    x=find(x),y=find(y);
    fa[x]=y;
    dis[x]+=num[y];
    num[y]+=num[x];
    num[x]=0; 
}
int main()
{
    init();
    int p,x,y;char op[3];
    scanf("%d",&p);
    while(p--)
    {
        scanf("%s",op);
        if(op[0]=='M')
        {
            scanf("%d%d",&x,&y);
            unite(x,y);
        }
        else
        {
            scanf("%d",&x);
            find(x);
            printf("%d\n",dis[x]);
        }
    }
    return 0;
}