在python中偷看堆

Question

在heapq库创建的python堆中偷看的官方方式是什么？现在我有

def heappeak(heap): 
    smallest = heappop(heap) 
    heappush(heap, smallest) 
    return smallest 

这是可以说，不是很好。我能否始终认为heap[0]是堆的顶部并使用它？或者会假设太多的底层实现？

Answer 1

是的，你可以做这样的假设，因为它是在documentation说：

堆是数组为其heap[k] <= heap[2*k+1]和heap[k] <= heap[2*k+2]所有ķ，从零开始计数 元素。为了比较 ，不存在的元素被认为是无限的 。 堆的有趣属性是 heap[0]始终是其最小的 元素。

（而这有可能是没有peek功能的原因：没有必要为它）

Answer 2

如果你正在使用Python 2.4或更新版本，你也可以使用heapq.nsmallest（） 。