POJ 1988Cube Stacking

一开始这题完全没有想法，后来又看题解，不懂...慢慢磨了两天左右，终于写出来了。

并查集的题，由于要输出当前标号方块下面的方块数，需要处理下。每次更新当前根节点下方块数目P为目标栈中方块数R，更新目标栈中方块数，合并栈。

查询时找出当前方块所有根节点的下方方块数P，记录下总和即为输出数据，同时将查询点根节点直接更新为当前栈最下方的根节点。当前点下方方块数目为输出的总和。此处不更新将超时

附代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define max(a,b) ((a)>(b)?(a):(b))
int root[30005];
int pos[30005];
int r[30005];
void init (int n)
{
    int i;
    for(i=0; i<n; i++)
    {
        root[i]=i;
        pos[i]=0;
        r[i]=1;
    }
}
int find(int x)
{
    if(x==root[x])return x;
    else return find(root[x]);
}
int main()
{
    int n,x,y,sum,i;
    char c[2];
    while(scanf("%d",&n)!=EOF)
    {
        init (30004);
        while(n--)
        {
            scanf("%s",c);
            if(c[0]=='M')
            {
                scanf("%d%d",&x,&y);
                pos[find(x)]=r[find(y)];
                r[find(y)]=r[find(y)]+r[find(x)];
                root[find(x)]=find(y);
                
            }
            else
            {
                scanf("%d",&x);
                sum=0;
                i=x;
                while(root[x]!=x)
                {
                    sum=sum+pos[x];
                    x=root[x];
                }
                root[i]=x;
                pos[i]=sum;
                printf("%d\n",sum);
            }
        }
    }
    return 0;
}