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

最新推荐文章于 2024-08-05 11:36:30 发布

CONDIMENTTTT

最新推荐文章于 2024-08-05 11:36:30 发布

阅读量227

点赞数 2

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/Caveira271_/article/details/137427560

版权

文章描述了一种算法，通过并查集数据结构处理无向图中的边添加(Cab)、连通性查询(Q1ab)和连通块大小查询(Q2a)，以判断两点是否在同一连通块以及计算某个点所在连通块的节点数量。

摘要由CSDN通过智能技术生成

给定一个包含 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}$

输入样例：

输出样例：

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;
}