[luoguP2342] 叠积木（并查集）

传送门

up[i] 表示一个木块上面有多少个

all[i] 表示整个连通块内有多少个

那么 一个木块下面的木块个数为 all[root[i]] - up[i] - 1

注意：up[i] 可以在 find 函数中维护，而 all[i] 不好维护，那么我们只需要祖先节点的 all[i] 表示整个连通块内木块的数目即可

　　　合并时也注意维护

——代码

#include <cstdio>
#include <iostream>
#define N 1000001

int n;
int f[N], up[N], all[N];

inline int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -1;
    for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
    return x * f;
}

inline int find(int x)
{
    if(x ^ f[x])
    {
        int fx = f[x];
        f[x] = find(f[x]);
        up[x] += up[fx];
    }
    return f[x];
}

int main()
{
    int i, x, y, fx, fy;
    char s[1];
    n = read();
    for(i = 1; i <= n; i++) f[i] = i, all[i] = 1;
    for(i = 1; i <= n; i++)
    {
        scanf("%s", s);
        if(s[0] == 'M')
        {
            x = read();
            y = read();
            fx = find(x);
            fy = find(y);
            f[fy] = fx;
            up[fy] += all[fx];
            all[fx] += all[fy];
        }
        else
        {
            x = read();
            fx = find(x);
            printf("%d
", all[fx] - up[x] - 1);
        }
    }
    return 0;
}