一个简单的关于堆的示例代码，不知道为什么竟是这个结果。。。。。。。。。。。。

import heapq

heap = [(2.7, 1), (1.7, 2), (0.7, 3), (0.3, 4), (1.3, 5)]
output = []
for _ in range(2):
    output.append(heapq.heappop(heap)[1])
print(output)


这段代码输出结果是[1, 3]。我是上网查了半天，说是heappop崩出来的是最小的值，那么不应该是0.3 和 0.4那两对元组吗？

所以最后结果应该是[3, 4]啊。这个1是从哪来的？

就是这个heap里怎么算最小值，都没有(2.7, 1)这个元组什么事啊。按第一个排取最小，也轮不到2.7；按第二个排取最小，那最后应该是2.7和1.7这两对元组的。

求解惑

Heapify 了么？

reddest1001 发表于 2021-10-18 21:24
Heapify 了么？

原题是272 Closest Binary Search Tree Value II。我抽象出来问题了。原代码如下，并没有用到heapify
class Solution:
    def closestKValues(self, root: Optional[TreeNode], target: float, k: int) -> List[int]:
        
        def dfs(root, heap):
            
            if not root:
                return
            
            dfs(root.left, heap)
            
            heapq.heappush(heap, (abs(root.val-target), root.val))
            
            dfs(root.right, heap)
        
        heap = []
        dfs(root, heap)
        
        result = []
        for _ in range(k):
            
            result.append(heapq.heappop(heap)[1])
        
        return result

以题目例1为例，target = 3.7, k = 2。那么中序遍历以后heap就是我上述的heap，然后后续代码运行完结果应该是[3, 4]。可是结果[1, 3]竟然运行通过。

在for循环以前加入
heapq.heapify(heap)就对了
正确的结果也不是[3,4]，而是[4,3]

reddest1001 发表于 2021-10-18 21:31
在for循环以前加入
heapq.heapify(heap)就对了
正确的结果也不是[3,4]，而是[4,3]

我的意思是上面那个272题目完整代码，没加heapify也运行通过了。然而不加的运行结果并不是[4, 3]，而是[1, 3], 就像我写在主题的那个抽象代码那样

ATPtennis 发表于 2021-10-18 14:58
我的意思是上面那个272题目完整代码，没加heapify也运行通过了。然而不加的运行结果并不是[4, 3]，而是[1 ...

heapq push进去的东西永远能保持是一个heap

但是你开始那个例子，把list直接当成heap用，那么就需要heapify一把，然后才能pop。。。

reddest1001 发表于 2021-10-19 01:01
heapq push进去的东西永远能保持是一个heap

但是你开始那个例子，把list直接当成heap用，那么就需要he ...

啊！好想明白点了，还得补一下基础知识！！谢谢！！！！！！