算法基础之数据结构——并查集2

题目：连通块的数量
给定一个包含 n 个点（编号为 1∼n）的无向图，初始时图中没有边。

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

C a b，在点 a 和点 b 之间连一条边，a 和 b 可能相等；
Q1 a b，询问点 a 和点 b 是否在同一个连通块中，a 和 b 可能相等；
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≤10^5
输入样例：
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 = 100005;

int p[N], num[N];//p为记录所在树的上一个节点
//find函数是为了找到集合所在树的根节点位置
int find(int x)
{
    if(p[x] != x)   p[x] =  find(p[x]);
    else    return x;
}
//deal函数是将两个集合链接到一起
void deal(int a, int b)
{
    if(find(a) != find(b)){
        int t = num[find(a)];//存储a开始所在节点的整个集合的元素个数
        num[find(a)] = 0;
        p[find(a)] = find(b);
        num[find(b)] += t;//将两个集合的元素个数相加
    }
}

bool query(int a,int b)
{
    if(find(a) == find(b))  return true;
    else    return false;
}

int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 0; i <= n; i ++)    p[i] = i, num[i] = 1;
    
    while(m --){
        string ins;
        int a, b;
        cin >> ins ;
        if(ins == "C"){
            cin >> a >> b;
            deal(a, b);
        }
        
        else if(ins == "Q1"){
            cin >> a >> b;
            if(query(a, b)) cout<<"Yes"<<endl;
            else    cout<<"No"<<endl;
        }
        
        else{
            cin >> a;
            cout << num[find(a)]<<endl;
        }
    }
}