11.连通块中点的数量 并查集

 

 用并查集维护每个集合中的元素个数

还是用集合来维护连通块

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int p[N]; //存储每个元素父节点是谁
int se[N]; //记录每一个集合中点的数量
//规定只有根节点的size才有意义 
int find(int x) { //返回x的祖宗节点，即返回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;
        se[i] = 1; //开始时每个连通块中点的数量为1
    }
    while (m--) {
        string op;
        int a, b;
        cin >> op;
        if (op == "C") {
            cin >> a >> b;
            if (find(a) == find(b)) { //特判
                continue;
            }
            se[find(b)] += se[find(a)];
            p[find(a)] = find(b);
        } else if (op == "Q1"){
            cin >> a >> b;
            if (find(a) == find(b)) {
                cout << "Yes" << endl;
            } else {
                cout << "No" << endl;
            }
        } else {
            cin >> a;
            cout << se[find(a)] << endl;
        }
    }
    return 0;
}