poj1988 Cube Stacking（并查集

题目地址：http://poj.org/problem?id=1988

 

题意：共n个数，p个操作。输入p。有两个操作M和C。M x y表示把x所在的栈放到y所在的栈上（比如M 2 6:[2 4]放到[1 6]上为[2 4 1 6]）,C x为输出x下面有几个数。

 

思路：并查集每个集合以栈最下面的数为根，维护两个数组num[x]表示x所在集合节点总数，count[x]表示x下方节点个数。每次查找压缩路径的时候更新count（换父节点的时候每轮都把父节点的count加给儿子，就可以一直更新到x所在栈的最底下），合并的时候更新px的count和py的num（把x的栈放到y的栈上，x下面多了num[y]个节点，新栈总根y总数增加num[x]个）。【不管是M还是C，先found一下x，把x的fa[x]更新到栈的最底下，count[x]才是真正的数！】

 

#include <iostream>
#include <cstdio>

using namespace std;

const int maxn = 30000 + 5;
struct c
{
    int fa, count, num;
}cube[maxn];

void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        cube[i].fa = i;
        cube[i].num = 1;
        cube[i].count = 0;
    }
}

int found(int x)
{
    if(cube[x].fa == x)
        return x;
    int px = cube[x].fa;
    cube[x].fa = found(cube[x].fa);
    cube[x].count += cube[px].count;
    return cube[x].fa;
}

void unite(int x, int y)
{
    int px = cube[x].fa;
    int py = cube[y].fa;
    if(px != py)
    {
        cube[px].fa = py;
        cube[px].count += cube[py].num;
        cube[py].num += cube[px].num;
    }
}

int main()
{
    int p, x, y;
    char opr;
    init(maxn);
    scanf("%d", &p);
    while(p--)
    {
        getchar();
        scanf("%c", &opr);
        if(opr == 'M')
        {
            scanf("%d%d", &x, &y);
            int px = found(x);
            int py = found(y);
            if(px != py)
                unite(x, y);
        }
        else
        {
            scanf("%d", &x);
            found(x);
            printf("%d\n", cube[x].count);
        }
    }
    return 0;
}