hdu-2818-Building Block-带权并查集

http://acm.hdu.edu.cn/showproblem.php?pid=2818

有n条管pipe，能把砖块串起来，初始每条管穿了一个砖

题意： 有2种操作：

  M x y把x的管放在y管的上面

 C x求x的下面有多少个砖块

带权并查集，维护一个num[i]表示i到根节点之间有多少块砖，再维护一个tol[x]数组，表示管x上一共串有多少个砖块

合并就很简单了：

if (fx!=fy)
                {
                  fa[fx]=fy;
                  num[fx]=tol[fy];<span style="white-space:pre">	</span>
                  tol[fy]+=tol[fx];
                }

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;

const double pi=acos(-1.0);
double eps=0.000001;

int fa[31234];
int num[31234];
int tol[31234];

int find(int x)
{
    if (fa[x]==x)    return x;
    int tmp=find(fa[x]);
num[x]=num[x]+num[fa[x]];
     return fa[x]=tmp;

}
int main()
{

    int n,m,k;
    while(cin>>n)
    {
        int x,y,z;
        char s[3];
        for (int i=0; i<=31000; i++)fa[i]=i,num[i]=0,tol[i]=1;
        for (int i=1; i<=n; i++)
        {
            scanf("%s",s);
            if (s[0]=='M')
            {
                scanf("%d%d",&x,&y);
                int fx=find(x);
                int fy=find(y);
                if (fx!=fy)
                {
                  fa[fx]=fy;
                  num[fx]=tol[fy];
                  tol[fy]+=tol[fx];
                }
            }
            else
            {
                scanf("%d",&x);
                find(x);
                printf("%d\n",num[x ]);

            }
         }

    }
    return 0;

}