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
,rank
,under
;
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【】一下，使其更新，然后再输出。