Hdu 2818 Building Block 加权并查集

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2818&&http://poj.org/problem?id=1988

题目大意：有N个木块，有两种操作，第一种是M x y，即把X木块所在的堆所有木块都放到y木块所在的堆上，第二种操作C x，即查询x木块下面的木块数

解题思路：对于合并操作和查询操作，并查集是一种不错的选择，但是本题查询的内容为木块下面的个数，这就相当于多了一个合并关系，我们可以开一个数组用来记录这个关系，当然这个关系的更新需要在查找时路径压缩实现和合并时实现，具体代码如下：

#include<stdio.h>
#define N 30005
int f[N],rank[N],under[N];
void init()
{
    for(int i=0;i<N;i++)//不明白为什么i从1开始初始化就Wa，从0就A了，数据范围不是从1-N嘛
    {
        f[i]=i;
        rank[i]=1;
        under[i]=0;
    }
}
//加权并查集路径压缩时更新权值
int find(int x)
{
    int temp;
    if(f[x]==x) return x;
    temp=f[x];
    f[x]=find(temp);
    under[x]+=under[temp];
    return f[x];
}
void merge(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        f[fx]=fy;//最下层为根节点
        under[fx]=rank[fy];//把根节点下面的堆为ueder【fx】
        rank[fy]+=rank[fx];//把两堆合并为一堆
    }
}
int main()
{
    int p,a,b;
    char c[5];
    while(~scanf("%d",&p))
    {
        init();
        for(int i=1;i<=p;i++)
        {
            scanf("%s",c);
            if(c[0]=='M')
            {
                scanf("%d%d",&a,&b);
                merge(a,b);
            }
            else
            {
                scanf("%d",&a);
                find(a);//每次询问之前更新一次当前节点
                printf("%d\n",under[a]);
            }
        }
    }
    return 0;
}

对于under[]值的更新，我自己的理解是这样的：

首先，当两个木块1,2合并时，由于他们的父节点都是他们本身，故 在merge【】函数中更新，此时under【1】==rank【2】==1，under【2】==0，rank【2】+=rank【1】==2；

如果木块3和1合并时，先find【1】，然后更新under【1】的值，再更新under【3】==rank【2】==2；

如果两个集合合并，找到他们的根节点，把根节点合并，则被合并的根节点下面的子节点尚未被更新，故需在询问时再find【】一下，使其更新，然后再输出。