[AcWing 837] 连通块中点的数量

---

点击查看代码

#include<iostream>

using namespace std;
const int N = 1e5 + 10;
int p[N], sz[N];
int find(int x)
{
    if (p[x] != x)  p[x] = find(p[x]);
    return p[x];
}
int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n ; i ++) {
        p[i] = i;
        sz[i] = 1;
    }
    while (m --) {
        string op;
        int a, b;
        cin >> op;
        if (op == "C") {
            cin >> a >> b;
            if (find(a) == find(b)) continue;
            sz[find(b)] += sz[find(a)];
            p[find(a)] = find(b);
        }
        else if (op == "Q1") {
            cin >> a >> b;
            if (find(a) == find(b)) puts("Yes");
            else    puts("No");
        }
        else {
            cin >> a;
            cout << sz[find(a)] << endl;
        }
    }
    return 0;
}

---

1. sz 数组用来记录集合中的元素个数，只有根节点对应的 sz 值有意义；
2. 合并集合时，将 a 的根节点的 size 加到 b的根节点的 size，再把 a 和 b 合并；