百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

Python数据结构堆的实现

toyiye 2024-06-21 12:02 10 浏览 0 评论

在本文中,我们将了解 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.Heap[i] > self.Heap[2 * i]  or  self.Heap[i] > self.Heap[(2 * i) + 1]): 
                if self.Heap[2 * i] < self.Heap[(2 * i) + 1]:
     # Swap the node with the left child and then call the min_heapify function on it
                    self.swapnodes(i, 2 * i)
                    self.min_heapify(2 * i)
  
                else:
                # Swap the node with right child and then call the min_heapify function on it
                    self.swapnodes(i, (2 * i) + 1)
                    self.min_heapify((2 * i) + 1)
  
    # THE HEAPPUSH FUNCTION
    def heappush(self, 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()

相关推荐

为何越来越多的编程语言使用JSON(为什么编程)

JSON是JavascriptObjectNotation的缩写,意思是Javascript对象表示法,是一种易于人类阅读和对编程友好的文本数据传递方法,是JavaScript语言规范定义的一个子...

何时在数据库中使用 JSON(数据库用json格式存储)

在本文中,您将了解何时应考虑将JSON数据类型添加到表中以及何时应避免使用它们。每天?分享?最新?软件?开发?,Devops,敏捷?,测试?以及?项目?管理?最新?,最热门?的?文章?,每天?花?...

MySQL 从零开始:05 数据类型(mysql数据类型有哪些,并举例)

前面的讲解中已经接触到了表的创建,表的创建是对字段的声明,比如:上述语句声明了字段的名称、类型、所占空间、默认值和是否可以为空等信息。其中的int、varchar、char和decimal都...

JSON对象花样进阶(json格式对象)

一、引言在现代Web开发中,JSON(JavaScriptObjectNotation)已经成为数据交换的标准格式。无论是从前端向后端发送数据,还是从后端接收数据,JSON都是不可或缺的一部分。...

深入理解 JSON 和 Form-data(json和formdata提交区别)

在讨论现代网络开发与API设计的语境下,理解客户端和服务器间如何有效且可靠地交换数据变得尤为关键。这里,特别值得关注的是两种主流数据格式:...

JSON 语法(json 语法 priority)

JSON语法是JavaScript语法的子集。JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔花括号保存对象方括号保存数组JS...

JSON语法详解(json的语法规则)

JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔大括号保存对象中括号保存数组注意:json的key是字符串,且必须是双引号,不能是单引号...

MySQL JSON数据类型操作(mysql的json)

概述mysql自5.7.8版本开始,就支持了json结构的数据存储和查询,这表明了mysql也在不断的学习和增加nosql数据库的有点。但mysql毕竟是关系型数据库,在处理json这种非结构化的数据...

JSON的数据模式(json数据格式示例)

像XML模式一样,JSON数据格式也有Schema,这是一个基于JSON格式的规范。JSON模式也以JSON格式编写。它用于验证JSON数据。JSON模式示例以下代码显示了基本的JSON模式。{"...

前端学习——JSON格式详解(后端json格式)

JSON(JavaScriptObjectNotation)是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。它基于JavaScriptProgrammingLa...

什么是 JSON:详解 JSON 及其优势(什么叫json)

现在程序员还有谁不知道JSON吗?无论对于前端还是后端,JSON都是一种常见的数据格式。那么JSON到底是什么呢?JSON的定义...

PostgreSQL JSON 类型:处理结构化数据

PostgreSQL提供JSON类型,以存储结构化数据。JSON是一种开放的数据格式,可用于存储各种类型的值。什么是JSON类型?JSON类型表示JSON(JavaScriptO...

JavaScript:JSON、三种包装类(javascript 包)

JOSN:我们希望可以将一个对象在不同的语言中进行传递,以达到通信的目的,最佳方式就是将一个对象转换为字符串的形式JSON(JavaScriptObjectNotation)-JS的对象表示法...

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json简介:Json,全名JavaScriptObjectNotation,JSON(JavaScriptObjectNotation(记号、标记))是一种轻量级的数据交换格式。它基于J...

比较一下JSON与XML两种数据格式?(json和xml哪个好)

JSON(JavaScriptObjectNotation)和XML(eXtensibleMarkupLanguage)是在日常开发中比较常用的两种数据格式,它们主要的作用就是用来进行数据的传...

取消回复欢迎 发表评论:

请填写验证码