hdu 2818 Building Block（加权并查集）2009 Multi-University Training Contest 1

题意：

一共有30000个箱子，刚开始时都是分开放置的。接下来会有两种操作：

1. M x y，表示把x箱子所在的一摞放到y箱子那一摞上。

2. C y，表示询问y下方有多少个箱子。

 

输入：

首行输入一个整数m，表示一共有m次操作。

接下来每次操作都是上面所说的两种中的一种。

 

输出：

每次C y，输出一个结果，每次输出占1行。

 

就是不知道究竟是只有一组测试样例还是有多组测试样例，不过我还是按照多组的方式写了。

题解：

这道题我自己思考了好久都没有思路，最后中忍不住去看了题解，然后被题解震惊了。

明明只有1个权值，题解中的方式愣是把这一个权拆成了两部分，然后算出结果，Orz。

两个权是tot[], sum[]，tot[i]表示i所在的箱子堆中一共有tot[i]个箱子，其中只有根节点的tot[i]是必须的。sum[i]则表示i下方共有sum[i]个箱子。

接下来，每次合并时，都将被合并的根节点fx的sum[]值加上仍然存在的根节点fy的tot[]值，然后fy的tot[]则加上fx的tot[]。

即，sum[fx] += tot[fy]; tot[fy] += tot[fx];

接下来，每次查找时，都要讲节点x的sum[]加上x的父节点的sum[]。

即，sum[x] += sum[fm[x]];

 

具体见代码——

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int N = 30010;

int fm[N], sum[N], tot[N];
int n;
int a, b;
char s[2];

void init()
{
    for(int i = 0; i < N; i++)          //很坑爹，题目明明说是1~30000，但是必须将0也初始化才能过
    {
        fm[i] = i;
        sum[i] = 0;
        tot[i] = 1;
    }
}

int mfind(int x)
{
    if(x == fm[x]) return x;
    int t = fm[x];
    fm[x] = mfind(fm[x]);
    sum[x] += sum[t];
    return fm[x];
}

void mmerge()
{
    int fx = mfind(a);
    int fy = mfind(b);
    if(fx != fy)
    {
        fm[fx] = fy;
        sum[fx] += tot[fy];
        tot[fy] += tot[fx];
    }
}

void work()
{
    for(int tm = 1; tm <= n; tm++)
    {
        scanf("%s", s);
        if(s[0] == 'M')
        {
            scanf("%d%d", &a, &b);
            mmerge();
        }
        else
        {
            scanf("%d", &a);
            mfind(a);
            printf("%d\n", sum[a]);
        }
    }
}

int main()
{
    //freopen("test.in", "r", stdin);
    while(~scanf("%d", &n))
    {
        init();
        work();
    }
    return 0;
}