POJ 1988 Cube Stacking

题意：有n个方块栈，一开始每个栈里只有一个方块，方块编号为栈编号，两种操作，一种是把方块a所在的栈全放在方块b所在的栈上面，另一种要求输出方块a的下面有多少个方块。

解法：并查集。用数组size[i]维护以i为根的集合大小，数组ans[i]维护方块i下面有多少方块，那么当两个栈合并时，栈a的底的ans即为栈b的size，而两个栈合并后根还是栈b的栈底，所以栈b的size+=栈a的size；当查询一个方块a下面有多少方块时，ans[a]+=ans[father[a]]。

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<math.h>
#include<limits.h>
#include<time.h>
#include<stdlib.h>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<iomanip>
#define LL long long
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

using namespace std;

int size[30005], ans[30005], father[30005];
int Find(int a)
{
    if(father[a] != a)
    {
        int fa = father[a];
        father[a] = Find(father[a]);
        ans[a] += ans[fa];
    }
    return father[a];
}
void Union(int a, int b)
{
    int fa = Find(a), fb = Find(b);
    ans[fa] = size[fb];
    size[fb] += size[fa];
    father[fa] = fb;
}
int main()
{
    int q;
    scanf("%d", &q);
    for(int i = 0; i <= 30000; i++) size[i] = 1;
    for(int i = 0; i <= 30000; i++) father[i] = i;
    memset(ans, 0, sizeof ans);
    while(q--)
    {
        char op[2];
        int a, b;
        scanf("%s%d", op, &a);
        if(op[0] == 'M')
        {
            scanf("%d", &b);
            Union(a, b);
        }
        else
        {
            Find(a);
            printf("%d
", ans[a]);
        }
    }
    return 0;
}