基本概念
并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。
并查集的思想是用一个数组表示了整片森林(parent),树的根节点唯一标识了一个集合,我们只要找到了某个元素的的树根,就能确定它在哪个集合里。
假设要把N个元素归为若干的集合,并实现以下功能:
union(p,q) - 将 p 和 q 两个元素连接起来。
find(p) - 查询 p 元素在哪个集合中。
isConnected(p,q) - 查看 p 和 q 两个元素是否相连接在一起。
把每一个元素,看做是一个节点并且指向自己的父节点,根节点指向自己。
判断两个元素是否连接,只需判断根节点是否相同。
连接两个元素,只需找到他们对应的根节点,让根节点相连即可。
构建这种指向父节点的树形结构, 使用一个数组构建一棵指向父节点的树,parent[i] 表示 i 元素所指向的父节点。
路径压缩
并查集里的 find 函数里可以进行路径压缩,是为了更快速的查找一个点的根节点。对于一个集合树来说,它的根节点下面可以依附着许多的节点,因此,我们可以尝试在 find 的过程中,从底向上,如果此时访问的节点不是根节点的话,那么我们可以把这个节点尽量的往上挪一挪,减少数的层数,这个过程就叫做路径压缩。
参考文献:并查集路径压缩 | 菜鸟教程 (runoob.com)https://www.runoob.com/data-structures/union-find-compress.html
将路径压缩至一层的方法:
...
// 查找过程, 查找元素p所对应的集合编号
// O(h)复杂度, h为树的高度
int find(int p) {
//深度最浅路径压缩算法
if (p != parent[p])
parent[p] = find(parent[p]);
return parent[p];
}
...
参考例题
第一行输入元素的个数
之后的各行,“I c1 c2 ”表示将C1 C2两个节点连接起来,“C c1 c2”表示监测两节点是否可以联通。
#include <iostream>
using namespace std;
const int maxn = 10005;
int fa[maxn];
int N, x, y;
char ch;
inline void init() {
for(int i = 1; i <= N; ++i)
fa[i] = i;
}
int find(int x) {//查询+路径压缩 把沿途的每个节点的父节点都设为根节点
if(x!=fa[x]){
fa[x]=find(fa[x]);
}
return fa[x];
}
void Merge(int x, int y) {
fa[find(x)] = find(y);
}
int main() {
scanf("%d", &N);
init();
getchar();
scanf("%c", &ch);
while(ch != 'S') {
scanf("%d %d", &x, &y);
getchar();
if(ch == 'C') {
if(find(x) == find(y)) cout << "yes" << endl;
else cout << "no" << endl;
} else if(ch == 'I') Merge(x, y);
scanf("%c", &ch);
}
int ans = 0;
for(int i = 1; i <= N; ++i) {
if(fa[i] == i) ans++;
}
if(ans == 1) cout << "The network is connected." << endl;
else cout << "There are " << ans << " components." << endl;
return 0;
}