提问者:小点点

使用python在优先级队列中打破平局


我使用堆队列来实现一个算法,当我添加新节点到我的队列,他们排序的启发式函数:例如heappush(队列,(分数(节点),节点),这是太棒了,除了事实,当我来弹出队列的下一个节点,我想要最近添加的节点,而不是第一个添加的节点,这是heappop返回。

我想我可以从第一个元素开始迭代,当下一个元素具有相同的分数时,继续。然后当我找到具有该分数的最后一个元素时,我选择它并将其从列表中删除。这显然不是很有效,并且打破了优先级队列的时间复杂度?

我被困在这件事上,想不出办法。

谢谢。

编辑;按照建议使用计数器是行不通的(也许我误解了)

>>> queue = []
>>> heappush(queue, (2, 0, 'a'))
>>> heappush(queue, (3, -1, 'b'))
>>> queue
[(2, 0, 'a'), (3, -1, 'b')]
>>> heappush(queue, (2, -2, 'c'))
>>> queue
[(2, -2, 'c'), (3, -1, 'b'), (2, 0, 'a')]

现在队列的排序不正确,并且比“a”更糟糕的选项“b”被放在它之前。

编辑2:

搞什么鬼?

>>> heappop(queue)
(2, -2, 'c')
>>> queue
[(2, 0, 'a'), (3, -1, 'b')]
>>> 

共1个答案

匿名用户

一种非常简单的方法是添加一个包含时间戳或计数器值的中间项。例如:

>>> heap = []
>>> counter = itertools.count()
>>> for i in range(10):
...     heapq.heappush(heap, (i, -next(counter), str(i) + ' oldest'))
...     heapq.heappush(heap, (i, -next(counter), str(i) + ' newest'))
... 
>>> heapq.heappop(heap)
(0, -1, '0 newest')
>>> heapq.heappop(heap)
(0, 0, '0 oldest')
>>> heapq.heappop(heap)
(1, -3, '1 newest')
>>> heapq.heappop(heap)
(1, -2, '1 oldest')

正如agf提到的,这是heapq文档使用的方法,只使用负索引。(那里的实现还包括一个很好的任务删除和优先级更改例程。)