POJ 1988

#include<stdio.h>

#include<string.h>

int bot[30005];//以每个集合最底下的那个立方体作为根节点，初始化为自己本身

int sum[30005];//表示以该元素为父亲节点的元素的编号的增量

int cisum[30005];//根节点所在集合的立方体总数

int num[30005];//记录每个节点在其父亲节点的集合内的编号，注意是其父亲节点而不一定是根节点

int findbot(int a)//找到根节点并且压缩路径

{

    int zsum=0,j;

    int tmp=a;

    while(a!=bot[a])//找到根节点，并且计算一路来所有是别人父亲元素的sum和

    {

        a=bot[a];

        zsum+=sum[a];

    }

    while(tmp!=bot[tmp])//压缩路径

    {

        num[tmp]+=zsum;//将num数组加上以后所有父亲节点的sum

        sum[tmp]+=zsum;//更新本身作为别人父亲节点的sum

        j=tmp;

        tmp=bot[tmp];

        zsum-=sum[tmp];//减掉自己本身的sum因为自己本身的sum与自己已经没有关系，注意sum是作为别人的父亲别人应该加上的数目

        bot[j]=a;

    }

    return a;

}

void link(int a,int b)

{

    int tmpa,tmpb;

    tmpa=findbot(a);

    tmpb=findbot(b);

    bot[tmpa]=tmpb;

    num[tmpa]+=cisum[tmpb];

    sum[tmpa]=cisum[tmpb];

    cisum[tmpb]+=cisum[tmpa];

}

int main()

{

    int n,a,b;

    char ca;

    scanf("%d",&n);

    {

        memset(sum,0,sizeof(sum));

        for(int i=1;i<=30001;i++)

        {

            bot[i]=i;

            cisum[i]=num[i]=1;

        }

        for(int i=0;i<n;i++)

        {

            getchar();

            scanf("%c",&ca);

            if(ca=='M')

            {

                scanf("%d%d",&a,&b);

                link(a,b);

            }

            else if(ca=='C')

            {

                scanf("%d",&a);

                findbot(a);

                printf("%d\n",num[a]-1);

            }

        }

    }

    return 0;

}