【数据结构】并查集

目录

并查集的概念

并查集的基本操作

并查集的表示

初始化

查询

合并

并查集的优化


并查集的概念

并查集就是拥有合并和查询功能的集合的集合。现在假设有下面三个集合:

{1、5、6、2},{3、4},{7、9、8}

现在从每个集合中选定一个元素作为这个集合的标志,假设这里选取1、3、7为标志,其他元素都挂靠在这个标志底下,使用树形结构进行存储,如下图所示:

这里需要注意他们没有实际的父子关系,只是同在一个集合里。

那么并查集需要做什么呢?查询操作指的是给定一个x,能够找到x所属集合的标志节点,即根节点。这里以2为例,其集合的标志节点为1;而合并操作指的是给定两个集合,需要把其中一个集合并到另一个集合里,即把一个集合的根节点连到另一个集合的根节点。合并7、3集合如下所示:

并查集的基本操作

并查集的表示

由上可以知道并查集的操作针对的元素主要是每个集合的标志节点,即根节点,所以使用树的双亲表示法来表示并查集会显得很方便。现在规定S[]数组用于存储每个节点的双亲,S[根节点]=集合中元素数量的相反数,那么就有下图所示的存储结构:

代码表示如下:

cpp 复制代码
#define SIZE 50
int UFsets[SIZE];

初始化

初始化定义一个循环把所有独立元素的UFsets都初始化为-1。

cpp 复制代码
void Initial(int S[]){
    for(int i=0;i<SIZE;i++)
        S[i]=-1;
}

查询

cpp 复制代码
int Find(int S[],int x){
    while(S[x]>=0)            //循环向上查找根节点
        x=S[x];
    return x;                //返回根节点
}

合并

cpp 复制代码
void Union(int S[],int Root1,int Root2){
    if(Root1==Root2)             //同属一个集合则返回
        return;
    S[Root1]=Root2;              //将一个根连到另一个下面
}

并查集的优化

上面提到的简单并查集查询方法最坏情况下时间复杂度为O(n),如何改进?方法有二,第一是优化合并操作,第二是优化Find操作。

对于Union的优化,如果把小树(即集合元素的数量比较小的树)连到大树的根上,显然会比把大树连到小树根上提高查询的效率,代码如下:

cpp 复制代码
void Union(int S[],int Root1,int Root2){
    if(S[Root1]=S[Root2])
        return;
    if(S[Root1]>S[Root2]){                    //第一棵树节点更少(注意这里是负数)
        S[Root2]+=S[Root1];                 //叠加节点数
        S[Root1]=Root2;                     //归根
    }
    else{
        S[Root1]+=S[Root2];                 //叠加节点数
        S[Root2]=Root1;                       //归根
    }
}

由数学归纳法可知这种方法造出的树深度不超过(不是很有必要知道推导过程)。

而对于Find的优化,可以知道如果所有的集合都只有两层(根节点和其他节点),那么查询的时间复杂度会大大下降。所以可以写一个循环把根节点到给定节点路径上的所有节点都归到根节点底下,优化代码如下:

cpp 复制代码
int Find(int S[],int x){
    int t;
    int root=x;                    
    while(S[root]>=0)        //找到根
        root=S[root];
    while(x!=root){          //压缩路径
        t=S[x];
        S[x]=root;
        x=t;
    }
    return root;
}

通过 Find 操作的"压缩路径"优化后,可使集合树的深度不超过 O(),其中 是一个增长极其缓慢的函数,对于常见的正整数 n,通常 <4。

相关推荐
森焱森1 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack3 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客4 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠4 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie988944 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼4 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表
.30-06Springfield5 小时前
决策树(Decision tree)算法详解(ID3、C4.5、CART)
人工智能·python·算法·决策树·机器学习
我不是哆啦A梦5 小时前
破解风电运维“百模大战”困局,机械版ChatGPT诞生?
运维·人工智能·python·算法·chatgpt
xiaolang_8616_wjl5 小时前
c++文字游戏_闯关打怪
开发语言·数据结构·c++·算法·c++20
small_wh1te_coder5 小时前
硬件嵌入式学习路线大总结(一):C语言与linux。内功心法——从入门到精通,彻底打通你的任督二脉!
linux·c语言·汇编·嵌入式硬件·算法·c