python的堆栈库heapq

Heapq库

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

heapq.heappush(heap, item)

将 item 的值加入 heap 中，保持堆的不变性。

heapq.heappop(heap)

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

heapq.heapify(x)

将list x 转换成堆，原地，线性时间内。

heapq.heapreplace(heap, item)

弹出并返回 heap 中最小的一项，同时推入新的 item。 堆的大小不变。 如果堆为空则引发 IndexError。

这个单步骤操作比 heappop() 加 heappush() 更高效，并且在使用固定大小的堆时更为适宜。 pop/push 组合总是会从堆中返回一个元素并将其替换为 item。

返回的值可能会比添加的 item 更大。 如果不希望如此，可考虑改用 heappushpop()。 它的 push/pop 组合会返回两个值中较小的一个，将较大的值留在堆中

import heapq

class Solution:
def lastStoneWeight(self, arry):
stone = [-i for i in arry]
heapq.heapify(stone)
while len(stone) > 1:
a = heapq.heappop(stone)
b = heapq.heappop(stone)
if a < b:
heapq.heappush(stone, a - b)
if not stone:
return 0
return -stone[0]