百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

Python数据结构:堆的实现(python数据结构教程)

itomcoil 2025-03-29 18:46 24 浏览

在本文中,我们将了解 Python 中的堆是什么以及怎样实现它。我们将通过最小堆的 python 程序实现来理解堆的概念。最后,我们将学习堆数据结构的时间复杂度和应用。那么,让我们开始吧!

什么是堆?

堆是一种遵循“完全”二叉树属性并满足堆属性的数据结构。因此,它也被称为二叉堆。完全二叉树是每一层都被填满,并且所有节点都尽可能靠左的树。在二叉树中,有可能最后一层是空的并且没有被填充。在堆数据结构中,我们为树的每个节点分配键值或权重。将根节点键值与子节点进行比较,然后根据比较大小将树相应地排列为两类,即最大堆和最小堆。堆数据结构可以用作堆排序算法来对数组或列表中的元素进行排序。堆排序算法可用于优先队列、订单统计、Prim 算法或Dijkstra 算法等。简而言之,堆数据结构在要重复删除最高或最低优先级对象时经常被使用。

建堆-Heapify?

首先我们需要了解什么是 heapify。使用二叉树创建堆数据结构的过程称为 Heapify。heapify 过程用于创建 Max-Heap 或 Min-Heap。让我们使用下面的示例来研究 Heapify:

考虑如下图所示的输入数组:

使用这个数组,我们将创建完整的二叉树。 我们从最后一个非叶子节点 (len(array)//2-1) 开始,将其作为当前的节点。如果要创建Min-Heap,我们要保证任何当前节点小于他的两个子节点。设当前节点的序号是k,那么其左子节点的序号是2k+1,右子节点是2k+2。Heapify就是要保证上述的局部性质,首先完成父节点的heapify,还要沿着一条树的路径递归完成子节点的heapify。接下来就是倒着数组序号进行Heapify,这样就完成了整个数组的堆化。数组在堆化过程中是以如下方式变化的:[3, 9, 2, 1, 4, 5]--> [3, 1, 2, 9, 4, 5]--> [3, 1, 2, 9, 4, 5]--> [1, 3, 2, 9, 4, 5]。以下程序演示了怎样heapify一个数组。时间复杂度是O(nlogn)

def min_heapify(A,k):
    print(A)
    l = left(k)
    r = right(k)
    if l < len(A) and A[l] < A[k]:
        smallest = l
    else:
        smallest = k
    if r < len(A) and A[r] < A[smallest]:
        smallest = r
    if smallest != k:
        A[k], A[smallest] = A[smallest], A[k]
        min_heapify(A, smallest)

def left(k):
    return 2 * k + 1

def right(k):
    return 2 * k + 2

def build_min_heap(A):
    n = int((len(A)//2)-1)
    for k in range(n, -1, -1):
        min_heapify(A,k)

A = [3,9,2,1,4,5]
build_min_heap(A)

理解min-heapify函数

此函数可以将节点及其所有后代(子节点及其子节点)遵循堆属性。它通过交换节点的键值来重新组织堆里的数据,使得当前节点成为其子树中的最小节点,遵循堆属性。

该函数首先在给定节点及其子节点中找到具有最小值的节点。然后它将给定节点(比如 i)与找到的最小值节点(比如 j)交换,然后在节点 j 上(递归地)调用 min-heapify 函数,以确保分配给节点 j 的新值确实不要破坏其子树中的堆属性。由于最多要遍历树的深度,所以它的时间复杂度是O(d),其中d是深度,或者,就节点数而言,O(log n),n是堆中的元素。

退出堆顶元素:heappop函数

该函数弹出堆的最小值(根元素)。

这实际上是通过将根节点与最后一个节点交换并删除现在的最后一个节点(包含最小值)然后为根节点调用 min-heapify 以在由于交换引起的更改后维护堆属性来完成的。

由于我们只需要调用一次min-heapify,因此时间复杂度为 O(log n),其中 n 是元素的数量,或者 O(h),其中 h 是树的高度,即 log n。

加入新元素:heappush 函数

此函数将一个新元素推入堆中,并将其排列到正确的位置,同时保持堆属性。

这实际上是通过在堆的末尾添加一个新节点来完成的。现在为了维护堆属性,我们从最后一个节点向上遍历(并在需要的地方交换)以修复可能被违反的堆属性。

与 heappop 类似,这里的时间复杂度是 O(log n),因为我们只需要遍历子树的高度。

获得最小值:extractMin 函数

此函数从堆中返回最高优先级(根元素)。由于我们只需要返回根的值而不对堆进行任何更改,并且根在 O(1) 时间内可以访问,因此函数的时间复杂度为 O(1)。

import sys
 
#defining a class min_heap for the heap data structure
 
class min_heap: 
    def __init__(self, sizelimit):
        self.sizelimit = sizelimit
        self.cur_size = 0
        self.Heap = [0]*(self.sizelimit + 1)
        self.Heap[0] = sys.maxsize * -1
        self.root = 1
 
    # helper function to swap the two given nodes of the heap
    # this function will be needed for heapify and insertion to swap nodes not in order
    def swapnodes(self, node1, node2):
        self.Heap[node1], self.Heap[node2] = self.Heap[node2], self.Heap[node1]
  
    # THE MIN_HEAPIFY FUNCTION
    def min_heapify(self, i):
  
        # If the node is a not a leaf node and is greater than any of its child
        if not (i >= (self.cur_size//2) and i <= self.cur_size: if self.heapi> self.Heap[2 * i]  or  self.Heap[i] > self.Heap[(2 * i) + 1]): 
                if self.Heap[2 * i] < self.heap2 i 1: swap the node with the left child and then call the min_heapify function it self.swapnodesi 2 i self.min_heapify2 i else: swap the node with right child and then call the min_heapify function it self.swapnodesi 2 i 1 self.min_heapify2 i 1 the heappush function def heappushself element: if self.cur_size>= self.sizelimit :
            return
        self.cur_size+= 1
        self.Heap[self.cur_size] = element 
        current = self.cur_size
        while self.Heap[current] < self.Heap[current//2]:
            self.swapnodes(current, current//2)
            current = current//2
  
    # THE HEAPPOP FUNCTION
    def heappop(self):
        last = self.Heap[self.root]
        self.Heap[self.root] = self.Heap[self.cur_size]
        self.cur_size -= 1
        self.min_heapify(self.root)
        return last
  
    # THE BUILD_HEAP FUNCTION
    def build_heap(self): 
        for i in range(self.cur_size//2, 0, -1):
            self.min_heapify(i)
  
  
    # helper function to print the heap
    def print_heap(self):
        for i in range(1, (self.cur_size//2)+1):
            print("Parent Node is "+ str(self.Heap[i])+" Left Child is "+ str(self.Heap[2 * i]) +                  " Right Child is "+ str(self.Heap[2 * i + 1]))
  
  
# Driver Code
minHeap = min_heap(10)
minHeap.heappush(15)
minHeap.heappush(7)
minHeap.heappush(9)
minHeap.heappush(4)
minHeap.heappush(13)
minHeap.print_heap()

相关推荐

MySQL修改密码_mysql怎么改密码忘了怎么办

拥有原来的用户名账户的密码mysqladmin-uroot-ppassword"test123"Enterpassword:【输入原来的密码】忘记原来root密码第一...

数据库密码配置项都不加密?心也太大了吧!

先看一份典型的配置文件...省略...##配置MySQL数据库连接spring.datasource.driver-class-name=com.mysql.jdbc.Driverspr...

Linux基础知识_linux基础入门知识

系统目录结构/bin:命令和应用程序。/boot:这里存放的是启动Linux时使用的一些核心文件,包括一些连接文件以及镜像文件。/dev:dev是Device(设备)的缩写,该目录...

MySQL密码重置_mysql密码重置教程

之前由于修改MySQL加密模式为mysql_native_password时操作失误,导致无法登陆MySQL数据库,后来摸索了一下,对MySQL数据库密码进行重置后顺利解决,步骤如下:1.先停止MyS...

Mysql8忘记密码/重置密码_mysql密码忘了怎么办?

Mysql8忘记密码/重置密码UBUNTU下Mysql8忘记密码/重置密码步骤如下:先说下大概步骤:修改配置文件,使得用空密码可以进入mysql。然后置当前root用户为空密码。再次修改配置文件,不能...

MySQL忘记密码怎么办?Windows环境下MySQL密码重置图文教程

有不少小白在使用Windows进行搭建主机的时候,安装了一些环境后,其中有MySQL设置后,然后不少马大哈忘记了MySQL的密码,导致在一些程序安装及配置的时候无法进行。这个时候怎么办呢?重置密码呗?...

10种常见的MySQL错误,你可中招?_mysql常见错误提示及解决方法

【51CTO.com快译】如果未能对MySQL8进行恰当的配置,您非但可能遇到无法顺利访问、或调用MySQL的窘境,而且还可能给真实的应用生产环境带来巨大的影响。本文列举了十种MySQL...

Mysql解压版安装过程_mysql解压版安装步骤

Mysql是目前软件开发中使用最多的关系型数据库,具体安装步骤如下:第一步:Mysql官网下载最新版(mysql解压版(mysql-5.7.17-winx64)),Mysql官方下载地址为:https...

MySQL Root密码重置指南:Windows新手友好教程

如果你忘记了MySQLroot密码,请按照以下简单步骤进行重置。你需要准备的工具:已安装的MySQL以管理员身份访问命令提示符一点复制粘贴的能力分步操作指南1.创建密码重置文件以管理员...

安卓手机基于python3搜索引擎_python调用安卓so库

环境:安卓手机手机品牌:vivox9s4G运行内存手机软件:utermux环境安装:1.java环境的安装2.redis环境的安装aptinstallredis3.elasticsearch环...

Python 包管理 3 - poetry_python community包

Poetry是一款现代化的Python依赖管理和打包工具。它通过一个pyproject.toml文件来统一管理你的项目依赖、配置和元数据,并用一个poetry.lock文件来锁定所有依赖的精...

Python web在线服务生产环境真实部署方案,可直接用

各位志同道合的朋友大家好,我是一个一直在一线互联网踩坑十余年的编码爱好者,现在将我们的各种经验以及架构实战分享出来,如果大家喜欢,就关注我,一起将技术学深学透,我会每一篇分享结束都会预告下一专题最近经...

官方玩梗:Python 3.14(πthon)稳定版发布,正式支持自由线程

IT之家10月7日消息,当地时间10月7日,Python软件基金会宣布Python3.14.0正式发布,也就是用户期待已久的圆周率(约3.14)版本,再加上谐音梗可戏称为π...

第一篇:如何使用 uv 创建 Python 虚拟环境

想象一下,你有一个使用Python3.10的后端应用程序,系统全局安装了a2.1、b2.2和c2.3这些包。一切运行正常,直到你开始一个新项目,它也使用Python3.10,但需要...

我用 Python 写了个自动整理下载目录的工具

经常用电脑的一定会遇到这种情况:每天我们都在从浏览器、微信、钉钉里下各种文件,什么截图、合同、安装包、临时文档,全都堆在下载文件夹里。起初还想着“过两天再整理”,结果一放就是好几年。结果某天想找一个发...