[AcWing]837. 连通块中点的数量(C++实现)并查集模板题
1. 题目
2. 读题(需要重点注意的东西)
思路:
简而言之,并查集查询两个值是否在一个集合的思想是看这两个值是否具有同一个根节点。
并查集合并的思路是将a集合直接接在b集合的根节点下。
优化的思路是路径压缩,即直接将集合中查询x的所有长辈节点直接接在根节点上。
该题只需要在并查集的基础上,加入一个cnt数组,来维护一个集合中节点的数量即可。即当加入一个集合时,将其根节点的cnt加到所加的根节点上。
3. 解法
---------------------------------------------------解法---------------------------------------------------
#include<iostream>
using namespace std;
const int N = 1e5+10;
int p[N],a,b;
int cnt[N];// 只保证根节点的cnt是有意义的,其含义是表明该集合的节点的数量
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;
cnt[i] = 1;
}
while(m--){
string op;
cin >> op;
if(op == "C"){
cin >> a >> b;
a = find(a);
b = find(b);
if(a != b){
p[a] = b;
cnt[b] += cnt[a]; // 当加入一个集合时,将其根节点的cnt加到所加的根节点上
}
}
else if(op == "Q1"){
cin >> a >> b;
find(a) == find(b) ? cout << "Yes" << endl : cout << "No" << endl;
}
else if(op == "Q2") {
cin >> a;
cout << cnt[find(a)] << endl;
}
}
return 0;
}
4. 可能有帮助的前置习题
5. 所用到的数据结构与算法思想
6. 总结
并查集模板题,推荐背下来。