描述
并查集是一种用于处理一些不相交集合的合并和查询问题的树型数据结构。通常在使用中以森林来表示。用森林的好处是降低了在合并(Union)操作时的时间复杂度。本文以并查集森林实现为例,学习该数据结构。
思路
- 用顺序表的形式来存储每个结点,每个结点的内容为该结点的直接父结点。例如,parent[i]表示第i个节点的父结点的索引。
- 为了使并(Union)操作时尽可能保持较低的树高度,新增一个size顺序表,size[i]表示以i为根结点所在的树中结点的个数。
- 初始化令每个结点的父结点为自身,相应的size都为1。
- 查找操作。对传入的索引进行查找,返回所在树的根节点的索引。就是一个向上遍历的过程,直到索引的父结点等于自身就为根节点。
- 合并操作。把树高度较低的那棵树的根节点指向另一棵树的根节点。
parent[pRoot] = qRoot
,两棵树合并为了一棵树。
实现代码
//UF1.h
#include<iostream>
#include<cassert>
using namespace std;
namespace UF1 {
class UnionFind {
private:
int *parent; //表示第i个元素的父结点
int *size; //表示以i为根的集合中的结点个数
int count; //结点总数
public:
UnionFind(int count) {
parent = new int[count];
size = new int[count];
this->count = count;
for(int i = 0; i < count; i++) {
parent[i] = i;
size[i] = 1;
}
}
~UnionFind() {
delete[] parent;
delete[] size;
}
int find(int p) {
while(p != parent[p]) {
p = parent[p];
}
return p;
}
bool isConnected(int p, int q) {
return find(p) == find