¿Cómo implementar colas de prioridad en Python?
o siento por una pregunta tan tonta, pero los documentos de Python son confusos .
Link 1: Implementación de cola http: //docs.python.org/library/queue.htm
Dice que esa cola tiene una construcción para la cola de prioridad. Pero no pude encontrar cómo implementarlo.
class Queue.PriorityQueue(maxsize=0)
Link 2: Implementación del montón http: //docs.python.org/library/heapq.htm
Aquí dicen que podemos implementar colas prioritarias indirectamente usando heapq
pq = [] # list of entries arranged in a heap
entry_finder = {} # mapping of tasks to entries
REMOVED = '<removed-task>' # placeholder for a removed task
counter = itertools.count() # unique sequence count
def add_task(task, priority=0):
'Add a new task or update the priority of an existing task'
if task in entry_finder:
remove_task(task)
count = next(counter)
entry = [priority, count, task]
entry_finder[task] = entry
heappush(pq, entry)
def remove_task(task):
'Mark an existing task as REMOVED. Raise KeyError if not found.'
entry = entry_finder.pop(task)
entry[-1] = REMOVED
def pop_task():
'Remove and return the lowest priority task. Raise KeyError if empty.'
while pq:
priority, count, task = heappop(pq)
if task is not REMOVED:
del entry_finder[task]
return task
raise KeyError('pop from an empty priority queue'
¿Cuál es la implementación de cola prioritaria más eficiente en Python? ¿Y cómo implementarlo?