Python heapq模块的常见方法

heapq – 堆队列

这个模块提供了堆队列算法的实现,也称为优先队列算法。

这里就不详细介绍堆的定义和特性了,有兴趣的读者可以访问 Python的Doc文档。heap-堆队列

要创建一个堆,可以使用list来初始化为 [] ,或者你可以通过一个函数 heapify() ,来把一个list转换成堆。

import heapq
heap = []
# 创建小根堆
heapq.heapify(heap)

#定义了以下函数:

heapq.heappush(heap, item)
# 将 item 的值加入 heap 中,保持堆的不变性。

heapq.heappop(heap)
#弹出并返回 heap 的最小的元素,保持堆的不变性。如果堆为空,抛出 IndexError .使用 heap[0],可以只访问最小的元素而不弹出它。

heapq.heappushpop(heap, item)
#将 item 放入堆中,然后弹出并返回 heap 的最小元素。该组合操作比先调用 heappush() 再调用 heappop() 运行起来更有效率。

heapq.heapify(x)
#将list x 转换成堆,原地,线性时间内。

heapq.heapreplace(heap, item)
#弹出并返回 heap 中最小的一项,同时推入新的 item。 堆的大小不变。 如果堆为空则引发 IndexError。

你可能感兴趣的:(数据结构与算法,数据结构)