[C++][算法基础]连通块中点的数量(并查集)

给定一个包含 n 个点（编号为 1∼n）的无向图，初始时图中没有边。

现在要进行 m 个操作，操作共有三种：

1. C a b，在点 a 和点 b 之间连一条边，a 和 b 可能相等；
2. Q1 a b，询问点 a 和点 b 是否在同一个连通块中，a 和 b 可能相等；
3. Q2 a，询问点 a 所在连通块中点的数量；

输入格式

第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 C a b，Q1 a b 或 Q2 a 中的一种。

输出格式

对于每个询问指令 Q1 a b，如果 a 和 b 在同一个连通块中，则输出 Yes，否则输出 No。

对于每个询问指令 Q2 a，输出一个整数表示点 a 所在连通块中点的数量

每个结果占一行。

数据范围

1≤n,m≤

输入样例：

5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5

输出样例：

Yes
2
3

代码：

#include<iostream>
using namespace std;

const int N = 100010;
int front[N],num[N];

int findleader(int x){
    if(front[x] != x){
        front[x] = findleader(front[x]);
    }
    return front[x];
}

int main(){
    int n,m;
    cin>>n>>m;
    for(int i = 0;i<n;i++){
        front[i] = i;
        num[i] = 1;
    }
    string op;
    int a,b,c;
    int pa,pb,pc;
    while(m--){
        cin>>op;
        if(op == "C"){
            cin>>a>>b;
            pa = findleader(a);
            pb = findleader(b);
            if(pa == pb){
                continue;
            }else{
                num[pb] += num[pa];
                front[pa] = pb;
            }
        }else if(op == "Q1"){
            cin>>a>>b;
            pa = findleader(a);
            pb = findleader(b);
            if(pa == pb){
                cout<<"Yes"<<endl;
            }else{
                cout<<"No"<<endl;
            }
        }else if(op == "Q2"){
            cin>>c;
            pc = findleader(c);
            cout<<num[pc]<<endl;
        }
    }
    return 0;
}