Python实现通用堆算法

完美代码

于 2023-06-29 01:24:01 发布

阅读量351

点赞数 1

CC 4.0 BY-SA版权

文章标签： python 算法开发语言

谢绝转载-https://update.blog.youkuaiyun.com

本文链接：https://blog.youkuaiyun.com/update7/article/details/131447812

Python. 专栏收录该内容

607 篇文章 ¥299.90 ¥399.90

订阅专栏

本文介绍了如何使用Python实现通用堆算法，包括最大堆和最小堆的概念，通过定义Heap类并实现入堆、出堆及调整堆结构的方法，展示了堆在排序中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python实现通用堆算法

堆是一种重要的数据结构，可用于排序、查找最大/小值等操作。本篇文章将介绍如何使用Python实现一个通用的堆算法，并附上完整的源代码。

首先我们需要了解堆的基本概念。堆是一种完全二叉树，通常分为两种情况：最大堆和最小堆。最大堆中，每个节点的值都大于等于它的子节点的值；最小堆中，每个节点的值都小于等于它的子节点的值。

接下来，我们定义一个名为Heap的类来表示堆：

class Heap:
    def __init__(self, cmp_func=None):
        self.cmp_func = cmp_func or (lambda x, y: x > y)
        self.arr = []

    def push(self, val):
        self.arr.append(val)
        self._shift_up(len(self.arr) - 1)

    def pop(self):
        if not self.arr:
            raise ValueError('Heap is empty')
        val = self.arr[0]
        last = self.arr.pop()
        if self.arr:
            self.arr[0] = last
            self._shift_down(0)

了解本专栏