poj1988 Cube Stacking（关系并查集）

http://poj.org/problem?id=1988

题意：给你30000个箱子，M  x y 代表把x所在的堆移动到y上，C x代表查询x号箱子下面的箱子数量。

思路：以每一堆最下面的箱子为根节点建树，below[N]代表某箱子下面的箱子数量，stackk[N]代表某一堆总共的箱子个数。Union操作只是将两堆根节点的相应参数变化，这样就合并成了一堆。箱子总数在递归过程中累加即可，重点是Find函数中below关系的处理。每次路径压缩将某个箱子的父节点改为根节点，其关系也要跟着变化。那当前的below值就等于父节点路径压缩前与当前节点的below值加上路径压缩后根节点与父节点的below值，每递归一次实现一次，递归真是神奇的东西啊！不过注意这里的Union操作只是改变根节点的值和参数，根节点上的参数并未改变，所以查询前要执行一次将值和参数更新。

#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <string.h>
#include <iostream>

using namespace std;

typedef long long LL;

const int N = 100010;
const int INF = 0x3f3f3f3f;

int pre[N], below[N], stackk[N];

int Find(int x)
{
    if(x == pre[x]) return x;
    int tmp = pre[x];
    pre[x] = Find(pre[x]);
    below[x] += below[tmp];
    return pre[x];
}

void Union(int x, int y)
{
    int rx = Find(x);
    int ry = Find(y);
    pre[rx] = ry;
    below[rx] += stackk[ry];
    stackk[ry] += stackk[rx];
}

int main()
{
   // freopen("in.txt", "r", stdin);
    int p, x, y;
    char s[5];
    scanf("%d", &p);
    for(int i = 1; i <= 30003; i++)
    {
        pre[i] = i;
        below[i] = 0;
        stackk[i] = 1;
    }
    while(p--)
    {
        scanf("%s", s);
        if(s[0] == 'M')
        {
            scanf("%d%d", &x, &y);
            Union(x, y);
        }
        else if(s[0] == 'C')
        {
            scanf("%d", &x);
            Find(x);
            printf("%d\n", below[x]);
        }
    }
    return 0;
}