在python中偷看一堆

Question

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

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

可以说,不是很好.我可以一直认为这heap[0]是堆的顶部并使用它吗？或者这会假设太多的底层实现？

Answer 1

是的,你可以做出这个假设,因为它在文档中说明:

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

(这可能是没有peek功能的原因:没有必要.)

Answer 2

如果您使用的是Python 2.4或更高版本,则还可以使用heapq.nsmallest().