【Python】heapq模块

菜菜2022

于 2021-08-11 16:12:16 发布

阅读量215

点赞数

CC 4.0 BY-SA版权

分类专栏： Python学习笔记

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/119609551

Python学习笔记专栏收录该内容

51 篇文章

订阅专栏

本文介绍了Python的heapq模块，通过heappush和heappop函数实现堆排序。示例中展示了两种方法：直接使用heappush和heappop进行元素入堆和出堆，以及先使用heapify将列表转换为堆再进行排序。堆排序结果一致，均为[1, 2, 3, 5, 23, 54, 132]。

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

import heapq

# 第一种
"""
函数定义：
heapq.heappush(heap, item)
    - Push the value item onto the heap, maintaining the heap invariant.
heapq.heappop(heap)
    - Pop and return the smallest item from the heap, maintaining the heap invariant.
    If the heap is empty, IndexError is raised. To access the smallest item without popping it, use heap[0].
"""
nums = [2, 3, 5, 1, 54, 23, 132]
heap = []
for num in nums:
    heapq.heappush(heap, num)  # 加入堆
print(heap)
print(heap[0])  # 如果只是想获取最小值而不是弹出，使用heap[0]
print([heapq.heappop(heap) for _ in range(len(nums))])  # 堆排序结果
# out: [1, 2, 3, 5, 23, 54, 132]


# 第二种
nums = [2, 3, 5, 1, 54, 23, 132]
heapq.heapify(nums)
print([heapq.heappop(nums) for _ in range(len(nums))])  # 堆排序结果
# out: [1, 2, 3, 5, 23, 54, 132]