【Python】heapq模块

最新推荐文章于 2025-04-18 13:44:20 发布

原创最新推荐文章于 2025-04-18 13:44:20 发布 · 215 阅读

1 ·

CC 4.0 BY-SA版权

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]