并查集路径压缩

简介: 如何描述一个复杂的连接关系?如图,很容易判断紧邻的2个人关系,但中间的连接很多很乱,怎么判断出两个人的关系呢?并查集就是一种结构,通过保存节点以及节点上的标签,来判断这两个节点是否连接在一起。

如何描述一个复杂的连接关系?如图,很容易判断紧邻的2个人关系,但中间的连接很多很乱,怎么判断出两个人的关系呢?
并查集就是一种结构,通过保存节点以及节点上的标签,来判断这两个节点是否连接在一起。当两个节点绑定时,可以任选其中一个节点的标签,指定另一个节点。当判断两个节点是不是连接时,可以上溯节点的祖宗节点,如果祖宗节点相同,那么节点相连。此时,节点上的标签可理解为指向父亲节点。

并查集的并&查

1.节点列表

并查集中的节点只需要保存父亲节点的信息,那么线性结构字典、列表都可以。我们用一维数组,索引是自身id,值指向父亲。
初始化时每个节点指向自身

class UnionFind:
    def __init__(self,size):
        self.size = size
        self.parent = np.arange(size)
    def union(self,p,q): ##将两个节点连接在一起
    def isConnected(self,p,q): ## 判断两个节点是否相连

2.判断两个节点相连

当两个节点的祖宗节点相同时,两个节点就是连接节点。

    def find(self,p):
        assert p>=0 and p<self.size
        while (self.parent[p]!=p):
    ##向上遍历,当父节点不是自己时,
    ##那么还存在父节点,继续遍历。
            p = self.parent[p]
        return p
    def isConnected(self,p,q):
    ##当祖宗节点相同时,两个节点是连接节点
        return self.find(p)==self.find(q)

3.节点连接

当节点连接时,需要将2个节点的祖宗节点相连,可任选一个节点连接另一个节点。

def union(self,p,q):
    pRoot = self.find(p)
    qRoot = self.find(q)
    if pRoot == qRoot:
        return 
    else:
        ## 第一个节点祖宗节点指向第二节点的祖宗节点
        self.parent[pRoot] = qRoot

4.优化连接

第3小节,我们任意选择一个节点连接,这样的选择有问题。举例:一层和二层节点集合合并:

  • 如果二层节点的祖宗节点连接到一层节点上,那么就形成了一个三层节点集。
  • 另一种可能,一层节点连接到二层祖宗节点,新集还是二层。

层数越少,查询祖宗节点的代价越小,应让节点层数少的连接到层数高的。

class UnionFind:
    def __init__(self,size):
        self.size = size
        self.parent = np.arange(size)
        self.rank = np.ones(size) //记录该节点下面的层次,默认都是1层
    def unionByRank(self,p,q):
        assert p>=0 and p<self.size and q>=0 and q<self.size
        pRoot = self.find(p)
        qRoot = self.find(q)
        if pRoot == qRoot:
            return self
        else:
        ## rank小的,添加到rank大的,这样的合并不增加rank。rank节点向上遍历的步数。
            if(self.rank[pRoot] > self.rank[qRoot]):
                self.parent[qRoot] = pRoot
            elif(self.rank[qRoot] > self.rank[pRoot]):
                self.parent[pRoot] = qRoot
            else:
                self.parent[pRoot] = qRoot
                self.rank[qRoot] +=1
            return self

5.路径压缩

进一步优化,使每个节点直接指向它的祖宗节点。

    def findCompress2(self,p):
        if p!=self.parent(p):
            self.parent[p] = self.findCompress2(self.parent[p])           
        return self.parent[p]

通过递归调用,函数从某一点出发,上溯到祖宗节点,返回值传递祖宗节点。函数返回时,相当于祖宗节点向下遍历,对每一个节点父节点重新赋值。

总结:

本文两个重点:介绍了并查集和路径压缩;单向列表的反向遍历。

目录
相关文章
|
4月前
|
存储 缓存 算法
软考通关密钥:计算机系统核心原理全解剖——软件设计师必懂的底层逻辑
专为软考中级软件设计师打造,深入解析CPU架构、存储体系、进制转换、原码补码、浮点数、寻址方式、校验码、RISC/CISC、流水线、Cache、中断、I/O控制、总线及加密技术等核心知识点,结合真题剖析高频考点,构建计算机系统底层知识体系,提升应试与实践能力。
375 1
软考通关密钥:计算机系统核心原理全解剖——软件设计师必懂的底层逻辑
|
网络协议 算法 Java
【Java网络编程】网络编程概述、UDP通信(DatagramPacket 与 DatagramSocket)
【Java网络编程】网络编程概述、UDP通信(DatagramPacket 与 DatagramSocket)
301 3
|
存储 安全 Java
列Java表(Lists)详解
在 Java 中,`List` 是一种用于存储有序集合的接口,允许重复元素,并提供了多种操作方法,如添加、删除、获取等。`List` 接口是 Java 集合框架的核心部分,常见的实现包括 `ArrayList`(动态数组)、`LinkedList`(双向链表)和 `Vector`(线程安全的动态数组)。
439 11
|
存储 数据处理 UED
计算机随机存取存储器(RAM )
【8月更文挑战第4天】
5424 8
中缀表达式转后缀表达式(逆波兰式)
中缀表达式转后缀表达式(逆波兰式)
1416 0
|
数据采集 存储 搜索推荐
使用Python构建自定义搜索引擎:从数据抓取到索引与搜索
使用Python构建自定义搜索引擎:从数据抓取到索引与搜索
|
Kubernetes 云计算 开发者
云计算中的容器化技术:Docker与Kubernetes的实践
云计算中的容器化技术:Docker与Kubernetes的实践
1336 0
|
算法 搜索推荐 数据库
【数据结构】二叉树算法讲解(定义+算法原理+源码)
【数据结构】二叉树算法讲解(定义+算法原理+源码)
|
存储 人工智能 缓存
数据结构顺序表和链表(超详细)
数据结构顺序表和链表(超详细)
325 1
快速排序详解(递归实现与非递归实现)
快速排序详解(递归实现与非递归实现)
352 0