poj 1988 Cube Stacking(并查集)

题目来源：poj-1988
题目大意：
给出n个立方体，可以将立方体移到其他立方体上形成堆；有p个如下操作：
⑴M X Y； 将X堆移动到Y所在堆上面；
⑵C X ； 输出在X所在堆下面的堆的个数。
题目分析：
使用并查集来解决，关键在于如何存储和更新每个立方体，在并查集基础上，再添加两个数组，dis[]用来存放在x下面的立方体个数，sum[]用来存放x所在堆的立方体总个数。对每一堆立方体，用立方体堆底的元素表示根节点，并且只用堆底元素表示该堆的立方体总个数。dis为0表示在堆底。在查找函数中，向上查找根节点，并保存当前节点的父节点,记为t，找到根节点后依次向下更新节点的dis，sum值。sum[x]==0时，表示x不在堆底，所以将x的dis加上其父节点的dis值；sum[x]！=0时，表示x在堆底需进行三个操作：dis[x]+=sum[t];sum[t]+=sum[x]; sum[x]=0;在合并函数中，合并完后再对x,y执行一次查找，来更新对应堆的值，因为在下次合并的时候，可能未来得及更新。
AC代码：

#include<stdio.h>
#include<string.h>
#define MAX 30010
int per[MAX];       //存放父节点 
int dis[MAX];       //dis[x]表示x下面的立方体个数 
int sum[MAX];       //sum[x]表示x所在立方体总个数 
int n;
void init()         //初始化 
{
    for(int i=1;i<=MAX;i++)
    {
        per[i]=i;
        sum[i]=1;
        dis[i]=0;
    }
}
int find(int x)     //查找函数 
{
    if(x!=per[x])
    {
        int t=per[x];
        per[x]=find(per[x]);
        if(sum[x])      //sum[x]！=0时 
        {
            dis[x]+=sum[t]; //父节点的个数加到x上 
            sum[t]+=sum[x]; //总个数加到父节点上 
            sum[x]=0;       //每个堆只用最底的表示总个数 
        }
        else
        dis[x]+=dis[t];     //sum[x]==0,说明不是堆底，就加到x上继续查找 
    }
    return per[x];
}
void join(int x,int y)      //合并函数 
{
    int fx=find(x);
    int fy=find(y);
    per[fx]=fy;
    find(x);                //再次查找，防止未更新堆值 
    find(y);
}
int main()
{   
    init();
    scanf("%d",&n);
    getchar();
    char ch;
    int x,y;
    for(int i=0;i<n;i++)
    {
        scanf("%c",&ch);
        getchar();
        if(ch=='M')             //输入'M'则连接两个堆 
        {
            scanf("%d%d\n",&x,&y);
//          getchar();          //用此吸收则输入老出错 
            join(x,y);
        }
        else                    //输入'C'则输出x所在堆下面的个数 
        {
            scanf("%d",&x);
            getchar();
            find(x);
            printf("%d\n",dis[x]);
        }
    }
    return 0;
}