837. 连通块中点的数量 知识点：并查集

837. 连通块中点的数量 - AcWing题库

刚开始只有五个点：

现在要进行m个操作： 1.把1与2连一条边：

即让1与2成为一个连通块

2.判断1与2是否在一个连通块

很明显在一个连通块，输出“Yes”

3.查询节点1所在的连通块有多少个节点

很明显有2个节点，输出“2”

4.把2与5连一条边：

此时1,2,5是一个连通块了

5.查询节点2所在的连通块有多少个节点

很明显有3个节点，输出“3”.

怎么求查询节点所在的连通块有多少个节点？

答： 假设只有root节点保存集合的节点个数：

那么此时a,b两个集合合并，我只用把b的root节点加上a的root节点就求出两个集合合并之后的节点个数了：

code

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int p[N];
int Size[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;
        Size[i]=1; //刚开始所有点都自成环，每个连通块都只有一个点
    }
    
    char op[5];int a,b;
    for(int i=0;i<m;i++)
    {
        scanf("%s",op);
     
        if(op[0]=='C')  //在a和b之间连一条边，即让a,b处于同一个连通块
        {
            scanf("%d%d",&a,&b);
            if(find(a)==find(b))continue;  //如果在一块连通块就没必要进行 下面的操作了
            Size[find(b)]=Size[find(b)]+Size[find(a)];
            p[find(a)]=find(b);  //让两个点的root节点一样就在一个连通块了。
        }
        else
        {
            if(op[1]=='1')   //查询a,b是否在同一个连通块
            {
                scanf("%d%d",&a,&b);
                if(find(a)==find(b))puts("Yes");
                else puts("No");
            }
            else  //求a节点所在的连通块的节点个数
            {
                scanf("%d%d",&a);
                printf("%d\n",Size[find(a)]);
            }
        }
        
        
    }
    return 0;
}