Question

我仍然是Python的新手，但是我遇到了堆优先级队列的问题。这是我的 init （）， str （），add（）和我的sift_up（）方法：

def __init__(self):
    self.queue = []

def __str__(self):
    return str(self.queue)

def add(self, item):
    self.queue.append(item)
    self.sift_up(len(self.queue) - 1)

def sift_up(self, item):
    parent = (item - 1) // 2
    if parent >= 0 and self.queue[parent] > self.queue[item]:
        self.queue[parent], self.queue[item] = self.queue[item], self.queue[parent]
        self.sift_up(parent)

现在，当我向队列中添加项目时，它们会很好。说，我把它放到终端：

pq = PriorityQueue()
pq.add(1)
pq.add(2)
pq.add(45)
pq.add(4)
pq.add(41)
pq.add(5)

pq.__str__()

我得到的是'[1,2,5,4,41,45]'。所以它看起来只是sift_up（），它没有完全重新排序堆。

编辑：每当我向队列添加“1”时，似乎搞砸了。在这个例子中，我在每次添加后返回：

>>> pq.add(5)
[5]
>>> pq.add(53)
[5, 53]
>>> pq.add(531)
[5, 53, 531]
>>> pq.add(5131)
[5, 53, 531, 5131]
>>> pq.add(1)
[1, 5, 531, 5131, 53]
>>>

因此它需要[1]处的任何元素并将其放在队列的后面。我确信这是微不足道的，但对Python来说是新手，我似乎无法弄清楚为什么。再次，非常感谢任何帮助！感谢。

Answer 1

在您的示例数据[5, 53, 531, 5131]中，您在sift_up中表达的计算将如下所示：

# Append 1 to the end
--> [5, 53, 531, 5131, 1]

# The index for '1' is 4, so 'item' is 4.
# (4-1) // 2 = 1 (and 1 >= 0), so 'parent' is 1.
# The value at 'parent' is 53. 53 > 1 is true.
# So swap the value 53 with the value at the end of the list.
--> [5, 1, 531, 5131, 53]

# Now repeat, 'item' starts out at 1.
# The index at (1 - 1) // 2 = 0 (And 0 >=0) so 'parent' is 0.
# The value at index 0 is 5. 5 > 1 is true.
# So swap the value 5 with the value at 'item' (1) to get
--> [1, 5, 531, 5131, 53]

因此，此结果符合您编码sift_up的方式。

标准库的heapq.heapify函数也产生相同的东西：看起来这是优先级队列的正确行为：

In [18]: import heapq

In [19]: x = [5, 53, 531, 5131, 1]

In [20]: heapq.heapify(x)

In [21]: x
Out[21]: [1, 5, 531, 5131, 53]

Python堆优先级队列sift_up（）

1 个答案: