heapq基本用法

最新推荐文章于 2025-03-25 17:10:36 发布

田晨|ggss

最新推荐文章于 2025-03-25 17:10:36 发布

阅读量560

点赞数

分类专栏： ubuntu18.04操作文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_41403872/article/details/119037178

版权

heapq是Python标准库中的一个模块，用于提供堆队列算法，也称为优先队列。本文将介绍heapq的基本用法，包括插入元素、删除最小元素以及如何构建最小堆等操作，并通过实例来演示其在解决实际问题中的应用。

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

"""
从列表中找出最大的或最小的N个元素
堆结构(大根堆/小根堆)
"""
import heapq

if __name__ == '__main__':

    # heapq.heappush 参数1用于排序的空列表，参数二需要被排序的列表，键入对排列中，被排序
    # heapq.heappop(heap)弹出堆排序列表中一个元素
    nums = [2, 3, 5, 1, 54, 23, 132]
    heap = []
    for num in nums:
        heapq.heappush(heap, num)
    print(heap)#[1, 2, 5, 3, 54, 23, 132]
    print(heapq.heappop(heap))#1
    print(heap)#[2, 3, 5, 132, 54, 23]
    print(nums)#[2, 3, 5, 1, 54, 23, 132]

    #heapq.heapify 将列表转化为堆排序完成的列表，默认从小到大
    nums = [