题目链接:http://poj.org/problem?id=1988
题意:有n个箱子,初始时每个箱子单独为一列;
接下来有p行输入,M, x, y 或者 C, x;
对于M,x,y:表示将x箱子所在的一列箱子搬到y所在的一列箱子上;
对于C,x:表示求箱子x下面有多少个箱子;
要注意题意是将x所在列的箱子叠到y所在箱子的上面,如果直接模拟的话就是将x最末端的叶子节点当做y的根节点的父亲节点合并,不过那样的话不好压缩路径,不压缩路径的话会超时.......
换个思路:
用数组son存储当前节点的子树大小,用vis存储当前节点到根节点的距离,所求值即son(find(x))-vis[x]-1;
按照一般并查集的合并方法,令pre[find(y)]=find(x);
那合并后会有: vis[find(y)]=son[find(x)]
son[find(x)]+=son[find(y)]
代码:
#include <iostream>
#include <stdio.h>
#define MAXN 30001
using namespace std; int pre[MAXN], son[MAXN], vis[MAXN]; int find(int x){
if(pre[x]==x){
return x;
}
int temp=pre[x]; //***递归思想,temp为存储x改变根节点后的根节点的临时变量
pre[x]=find(pre[x]);
vis[x]+=vis[temp]; //***x到改变前根节点的距离即x到temp的距离加上temp到根节点的距离
return pre[x];
} void jion(int x, int y){
int px=find(x);
int py=find(y);
if(px!=py){
pre[py]=px;
vis[py]=son[px]; //***将x所在列放到y所在列上面后,find(y)到新合并后的根节点的距离即为合并前find(x)的子树的大小
son[px]+=son[py]; //***合并后find(x)的子树大小即为合并前find(x)与find(y)的子树大小的和
}
} int main(void){
int p;
scanf("%d", &p);
for(int i=; i<=MAXN; i++){
pre[i]=i;
son[i]=;
}
for(int i=; i<=p; i++){
char s[];
int x, y;
scanf("%s", s);
if(s[]=='M'){
scanf("%d%d", &x, &y);
jion(x, y);
}else{
scanf("%d", &x);
printf("%d\n", son[find(x)]-vis[x]-);//***注意这里并不是输出son(x),因为我们并没有求出每个节点的子树的大小
}
}
return ;
}