[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 = , f = ;
     char ch = getchar();
     for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -;
     for(; isdigit(ch); ch = getchar()) x = (x << ) + (x << ) + ch - '';
     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[];
     n = read();
     for(i = ; i <= n; i++) f[i] = i, all[i] = ;
     for(i = ; i <= n; i++)
     {
         scanf("%s", s);
         if(s[] == '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\n", all[fx] - up[x] - );
         }
     }
     return ;
 }