在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中,其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
并查集是一种树型的数据结构,用于处理一些不相交集合(Disjoint Sets)的合并及查询问题。常常在使用中以tree来表示。
用途
维护一个无向图的连通性,判断n个点m条边时最少加多少边可以连通所有点
判断在一个无向图中,两点间加边是否会产生环(最小生成树克鲁斯卡尔中有用到)
维护集合等操作
操作
(1)Union(Root1, Root2):把子集合Root2并入集合Root1中。要求这两个集合互不相交,否则不执行合并。
(2)Find(x):搜索单元素x所在的集合,并返回该集合的名字。
(3)UnionFindSets(s):构造函数,将并查集中s个元素初始化为s个只有一个单元素的子集合。
初始化
用数组来建立一个并查集,数组下标代表元素,下标对应的值代表父节点,全部初始化为-1,根节点为一个集合的元素个数,数组的长度为并查集的初始连通分量的个数。并查集要求各集合是不相交的,因此要求x没有在其他集合中出现过。算法如下:
//构造函数
UF(int size){ this->count = size; array = new int[size]; for(int i = 0 ; i < size ; i++){ this->array[i] = -1; } }
查找操作
返回能代表x所在集合的节点,通常返回x所在集合的根节点。这里的查找操作通常采用路径压缩的办法,即在查找过程中组不减小树的高度,把元素逐步指向一开始的根节点。这样下次再找根节点的时间复杂度会变成o(1)。如下图所示
算法如下:
//查找操作,路径压缩
int Find(int x){ if(this->array[x] < 0){ return x; }else{ //首先查找x的父节点array[x],然后把根变成array[x],之后再返回根 return this->array[x] = Find(this->array[x]); } }
并操作
将包含x,y的动态集合合并为一个新的集合。合并两个集合的关键是找到两个集合的根节点,如果两个根节点相同则不用合并;如果不同,则需要合并。
这里对并操作有两种优化:根节点存树高的相反数或者根节点存集合的个数的相反数,这两种方法统称按秩归并。通常选用第二种方法。
归并过程如下图:
算法如下:
//并操作,跟结点存储集合元素个数的负数
//通过对根结点的比较
void Uion(int root1, int root2){ root1 = this->Find(root1); root2 = this->Find(root2); if(root1 == root2){ return; }else if(this->array[root1] < this->array[root2]){ //root1所代表的集合的个数大于root2所代表集合的个数 //因为为存放的是元素个数的负数 this->array[root1] += this->array[root2]; this->array[root2] = root1; count--; }else{ this->array[root2] += this->array[root1]; this->array[root1] = root2; count--; } } }
实现方案
1 树结构(父指针表示法)
用这种实现方式,每个集合用一棵树表示,树的每一个节点代表集合的一个单元素。所有各个集合的全集合构成一个森林,并用树与森林的父指针表示法来实现。其下标代表元素名。第I个数组元素代表包含集合元素I的树节点。树的根节点的下标代表集合名,根节点的父为-1,表示集合中元素个数。
下面看一个例子:
全集合是S = {0,1,2,3,4,5,6,7,8,9},初始化每个元素自成为一个单元素子集合。(书上原图,感觉挺清晰的)
经过一段时间的计算,这些子集合并成3个集合,他们是全集合S的子集合:S1 = {0,6,7,8},S2= {1,4,9},S3 = {2,3,5}。则表示他们并查集的树形结构如下图:
上面数组中的元素值有两种含义:
(1)负数表示当前节点是树的根节点,负数的绝对值表示树中节点的个数,也即集合中元素的个数。
(2)正数表示其所属的树的根节点,由树形表示很容易理解,这也是树的父指针表示的定义。
经过上面对相关数据的组织,再回头来看并查集的3中核心操作是怎样依托于树来实现的:
(1)将root2并入到root1中,其实就可以直接把root2的数组元素(就是他的父节点)改成root1的名字(就是他所在的数组下标)。
下面的图表示了合并两个子集合的过程:
(2)查找x所属于的根节点(或者说是x所属于的集合),就可以一直找array[x],直到array[x]小于0,则证明找到了根(所在集合)。
下面的图示意了查找一个节点所属集合的过程:
(3)将整个集合初始化为单元素集合,其实就是建立树的父指针数组的过程,把数组元素全初始化为-1,也就表示了每个元素都各占一个集合。
有了上面的理论,代码也比较容易实现出来!下面给出了一个代码的实例:
/* *树结构构建并查集,其中树用父指针形式表示 */ #include <iostream> const int DefaultSize = 10; class UFSets { //集合中的各个子集合互不相交 public: UFSets(int sz = DefaultSize); //构造函数 (并查集的基本操作) ~UFSets() { delete[] parent; } //析构函数 UFSets& operator = (UFSets& R); //重载函数:集合赋值 void Union(int Root1, int Root2); //两个子集合合并 (并查集的基本操作) int Find(int x); //搜寻x所在集合 (并查集的基本操作) void WeightedUnion(int Root1, int Root2); //加权的合并算法 private: int *parent; //集合元素数组(父指针数组) int size; //集合元素的数目 }; UFSets::UFSets(int sz) { //构造函数,sz是集合元素的个数,父指针数组的范围0到sz-1 size = sz; //集合元素的个数 parent = new int[size]; //开辟父指针数组 for (int i = 0; i < size; i ++) { //初始化父指针数组 parent[i] = -1; //每个自成单元素集合 } } int UFSets::Find(int x) { //函数搜索并返回包含元素x的树的根 while (parent[x] >= 0) { x = parent[x]; } return x; } void UFSets::Union(int Root1, int Root2) { //函数求两个不相交集合的并,要求Root1与Root2是不同的,且表示了子集合的名字 parent[Root1] += parent[Root2]; //更新Root1的元素个数 parent[Root2] = Root1; //令Root1作为Root2的父节点 } void UFSets::WeightedUnion(int Root1, int Root2) { //使用节点个数探查方法求两个UFSets集合的并 int r1 = Find(Root1); //找到root1集合的根 int r2 = Find(Root2); //找到root2集合的根 if (r1 != r2) { //两个集合不属于同一树 int temp = parent[r1] + parent[r2]; //计算总节点数 if (parent[r2] < parent[r1]) { //注意比较的是负数,越小元素越多,此处是r2元素多 parent[r1] = r2; //r1作为r2的孩子 parent[r2] = temp; //更新r2的节点个数 } else { parent[r2] = r1; //... parent[r1] = temp; //... } } }
代码的注释比较详尽,我就不在赘言。但是有一个注意点我已经写在了下面!
当前并查集的改进!
的确,有一个极端的状况使得上面的树实现的并查集性能低下!问题原因在于,这里没有规定子集合并的顺序,更确切的说是子集一直在向同一个方向依附:
下面的图片展示了当Union(0,1),Union(1,2),Union(2,3),Union(3,4)执行完后的树的形状。
在这种极端情况下他编变成了一个单链表(退化的树),这样的话,用Find函数查找完所有的节点所归属的集合将会开销的时间复杂度为:O(n^2)。