hdu 2818 带权并查集

/*题目大意：
有N个砖头，编号为1～N，  然后有两种操纵，
第一种是M  x  y， 把x地点的那一堆砖头全部移动放到y地点的那堆的上方。
第二种操纵是C  x，  即查询x下面有几许个砖头，并且输出。
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=30002;
int p,fa[maxn],up[maxn],sum[maxn];
// up[i] 表示节点 i 之上有多少个节点
// sum[i] 表示节点i及之下有多少个节点
char op[2];
int find(int i)
{
if(fa[i]==i)return i;
int ans=fa[i];
fa[i]=find(fa[i]);
up[i]+=up[ans];//当前节点 加上原先的父节点个数
return fa[i];
}
void Union(int u,int v)
{
fa[v]=u;
up[v]=sum[u];
sum[u]+=sum[v];

}
int main()
{
//freopen("//media/学习/ACM/input.txt","r",stdin);
while(scanf("%d",&p)!=EOF)
{
int u,v;
for(int i=0;i<maxn;i++)fa[i]=i,sum[i]=1,up[i]=0;
while(p--)
{
cin>>op;
if(op[0]=='M')
{
scanf("%d%d",&u,&v);
int x=find(u);
int y=find(v);
if(x!=y)
{
Union(x,y);
}
}else
{
scanf("%d",&u);
int y=find(u);
printf("%d\n",sum[y]-up[u]-1);
}
}
}
return 0;
}