Sbirciare in un mucchio in python

Question

Qual è il modo ufficiale di sbirciare in un heap python come creato dalle librerie heapq? In questo momento ho

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

che è discutibilmente, non molto bello. Posso sempre presumere che lo heap[0] sia la cima dell'heap e usarlo? O questo presuppone troppa implementazione sottostante?

Answer 1

Sì, si può fare questa ipotesi, perché si afferma nella documentation:

Cumuli sono array per cui heap[k] <= heap[2*k+1] e heap[k] <= heap[2*k+2] per tutti k, contando elementi da zero. Per il confronto , gli elementi non esistenti sono considerati infiniti. La proprietà interessante di un heap è che heap[0] è sempre il suo più piccolo elemento .

(E questo è probabilmente la ragione non esiste una funzione peek:. Non v'è alcun bisogno di esso)

Answer 2

Se stai usando Python 2.4 o più recente, è possibile utilizzare anche heapq.nsmallest() .