（并查集）连通块中点的数量

相对于比较裸的并查集题目，这道题只是单纯在代码中维护了一些新的信息，

而这些信息正是题目里要求的，例如本题中要求求连通块中点的数量，那么则

需要在代码中添加一个变量用来计数，主要是用来统计以某个点的祖先节点

为代表的连通块中点的数量。

#include <iostream>
#include <cstring>
using namespace std;
const int N = 100010;
int fa[N], cnt[N];
int find(int x){
    if(x != fa[x]) fa[x] = find(fa[x]);
    return fa[x];
}
int main(){
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i ++){
        fa[i] = i;
        cnt[i] = 1;
    }
    while(m --){
        char op[5];
        int a, b;
        cin >> op;
        if(op[0] == 'C'){
            cin >> a >> b;
            if(find(a) == find(b)) continue;
            cnt[find(b)] += cnt[find(a)];//注意这两步不能写反，因为应该是先统计点的数量再连接两个集合
            fa[find(a)] = find(b);
        }else if(op[1] == '1'){
            cin >> a >> b;
            if(find(a) == find(b)) cout << "Yes" << endl;
            else cout << "No" << endl;
        }else if(op[1] == '2'){
            cin >> a;
            cout << cnt[find(a)] << endl;
        }
    }
    return 0;
}