查看: 249|回复: 6
收起左侧

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

|只看干货
ATPtennis | 显示全部楼层 |阅读模式
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   93% (45)
 
 
6% (3)    👎

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看附件。没有帐号?注册账号

x
本帖最后由 ATPtennis 于 2021-10-18 20:22 编辑

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这两对元组的。

求解惑

评分

参与人数 1大米 +3 收起 理由
14417335 + 3

查看全部评分


上一篇:189 rotate array如何想到三次翻转?
下一篇:动态规划刷的让人想quit
reddest1001 2021-10-19 03:24:23 来自APP | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   97% (356)
 
 
2% (9)    👎
Heapify 了么?

评分

参与人数 1大米 +1 收起 理由
14417335 + 1

查看全部评分

回复

使用道具 举报

 楼主| ATPtennis 2021-10-19 03:29:07 | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   93% (45)
 
 
6% (3)    👎
本帖最后由 ATPtennis 于 2021-10-18 21:30 编辑

原题是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]竟然运行通过。
回复

使用道具 举报

reddest1001 2021-10-19 03:31:12 | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   97% (356)
 
 
2% (9)    👎
在for循环以前加入
heapq.heapify(heap)就对了
正确的结果也不是[3,4],而是[4,3]

评分

参与人数 1大米 +1 收起 理由
14417335 + 1

查看全部评分

回复

使用道具 举报

 楼主| ATPtennis 2021-10-19 05:58:24 | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   93% (45)
 
 
6% (3)    👎
本帖最后由 ATPtennis 于 2021-10-18 23:59 编辑
reddest1001 发表于 2021-10-18 21:31
在for循环以前加入
heapq.heapify(heap)就对了
正确的结果也不是[3,4],而是[4,3]

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

使用道具 举报

reddest1001 2021-10-19 07:01:46 | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   97% (356)
 
 
2% (9)    👎
ATPtennis 发表于 2021-10-18 14:58
我的意思是上面那个272题目完整代码,没加heapify也运行通过了。然而不加的运行结果并不是[4, 3],而是[1 ...

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

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

使用道具 举报

 楼主| ATPtennis 2021-10-19 14:39:36 | 显示全部楼层
本楼: 👍   0% (0)
 
 
0% (0)   👎
全局: 👍   93% (45)
 
 
6% (3)    👎
reddest1001 发表于 2021-10-19 01:01
heapq push进去的东西永远能保持是一个heap

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

啊!好想明白点了,还得补一下基础知识!!谢谢!!!!!!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册账号
隐私提醒:
  • ☑ 禁止发布广告,拉群,贴个人联系方式:找人请去🔗同学同事飞友,拉群请去🔗拉群结伴,广告请去🔗跳蚤市场,和 🔗租房广告|找室友
  • ☑ 论坛内容在发帖 30 分钟内可以编辑,过后则不能删帖。为防止被骚扰甚至人肉,不要公开留微信等联系方式,如有需求请以论坛私信方式发送。
  • ☑ 干货版块可免费使用 🔗超级匿名:面经(美国面经、中国面经、数科面经、PM面经),抖包袱(美国、中国)和录取汇报、定位选校版
  • ☑ 查阅全站 🔗各种匿名方法

本版积分规则

>
快速回复 返回顶部 返回列表