poj 1988 Cube Stacking 并查集

#include <stdio.h>
#define maxn 40000
int p[maxn],sum[maxn],d[maxn];
int find(int x)
{
    if(p[x]==x) return x;
    else
    {
        int tmp=p[x];
        p[x]=find(p[x]);
        d[x]+=d[tmp];
        return p[x];
    }
}
int main()
{
    int num;
    char o;
    while(scanf("%d",&num)!=EOF)
    {
        int i,j;
        int u,v;
        int fu,fv;
        for(i=1;i<=33000;i++)
        {
            p[i]=i;
            sum[i]=1;
            d[i]=0;
        }
        getchar();
        for(i=1;i<=num;i++)
        {
            scanf("%c",&o);
            if(o=='C')
            {
                scanf("%d",&u);
                fu=find(u);
                printf("%d
",d[u]);
            }
            else
            {
                scanf("%d%d",&u,&v);
                fu=find(u);
                fv=find(v);
                p[fu]=fv;
                d[fu]+=sum[fv];
                sum[fv]+=sum[fu];
            }
            getchar();
        }
    }
    return 0;
}