poj1988(并查集)

题目链接：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 ;
 }