并查集
初步认识
- 顾名思义,并查集三个字拆开就是,“合并、查找、集合”。
- 实现形式,通过数组表示,其中数组的下标表示数据的值,数组的值表示指向的父亲节点。因此可得当
father[i] = i
的时候我们可以知道这表示的就是集合的根结点。
基本操作
- 初始化,每一个节点都作为一个独立的树
- 查找父节点,递归或者循环 终止条件
father[i] = i
- 合并, 合并两个树需要找到父节点在进行合并
典型例题
题目大意
input 1:
4 2
1 4
2 3
output 1:
2
input 2:
7 5
1 2
2 3
3 1
1 4
5 6
output 2:
3
题目分析
使用并查集进行操作,设置两个数组,flag表示本节点是否为根节点,father表示父亲的位置。
代码
#include<bits/stdc++.h>
using namespace std;
int m, n, sum;
vector<bool> flag;
vector<int> father;
void init(int n) {//初始化并查集地两个集合
flag.resize(n + 10);
father.resize(n + 10);
//对每一个表示是否是根地址的数组设置为true,父亲节点都指向自己
for (int i = 0; i < n; i++) {
flag[i] = true;
father[i] = i;
}
}
int findFather(int x) {//寻找本集合的父亲节点
int a = x;
//下标表示数据,属性值表示父亲节点
while (father[x] != x) {
x = father[x];
}
//新增路径压缩,目的是为了解决极端情况下查找根结点效率低下的问题
while (father[a] != a) {
int b = a;//防止修改
a = father[a];
father[b] = x;//所有的节点的父节点全部设置为根结点
}
return x;
}
void unionSet(int x, int y) {//合并两个集合
int rootX = findFather(x);
int rootY = findFather(y);
if (rootX != rootY) {
//属于不同的集合,此时进行合并操作
father[rootY] = rootX;
flag[rootY] = false;//删除根结点,此时flag置为false
}
}
int main() {
//n:数码宝贝个数、m:组数
cin >> n >> m;
//初始化并查集
init(n);
//读入数据
for (int i = 0; i < m; i++) {
int x, y;
cin >> x >> y;
//对同组数据进行合并操作
unionSet(x, y);
}
for (int i = 0; i < n; i++) {
if (flag[i] == true) {
sum++;
}
}
cout << sum;
return 0;
}