题解 | #输入n个整数,输出其中最小的k个#

输入n个整数,输出其中最小的k个

https://www.nowcoder.com/practice/69ef2267aafd4d52b250a272fd27052c

from queue import Queue  # 队列,FIFO
from queue import PriorityQueue  # 优先级队列,优先级高的先输出


def construct_min_heap(nums):
    n = len(nums)
    for i in range((n - 2) // 2, -1, -1):
        shift_down(nums, n, i)


def shift_down(nums: list, n: int, index: int):
    while index < n:
        val = nums[index]
        l_child_index, r_child_index = index * 2 + 1, index * 2 + 2
        l_child_val = 999999999 if l_child_index >= n else nums[l_child_index]
        r_child_val = 999999999 if r_child_index >= n else nums[r_child_index]
        if l_child_val == r_child_val == 999999999:
            break
        min_child_val = min(l_child_val, r_child_val)
        min_child_index = l_child_index if l_child_val < r_child_val else r_child_index
        if val > min_child_val:
            nums[index] = min_child_val
            nums[min_child_index] = val
            index = min_child_index
        else:
            break


def pop_min_heap(nums: list):
    ret = nums[0]
    nums[0] = nums.pop()
    shift_down(nums, len(nums), 0)
    return ret


n, k = [int(i) for i in input().split()]
nums = [int(i) for i in input().split()]
construct_min_heap(nums)
# print(nums)
for i in range(k):
    print(pop_min_heap(nums), end=" ")

全部评论

相关推荐

10-13 22:56
门头沟学院 C++
rt,鼠鼠的浪潮网签明天过期,鼠鼠是山东人,好像自己也能接受。之前的面试大厂基本挂干净了,剩下小米二面后在泡,问了下面试官没有挂,但要泡。还有海信似乎也通过了,不过在深圳,鼠鼠也不是很想去。其它还有一些公司应该陆陆续续还有一些面试,现在有些纠结是直接签了还是再等再面呢?大佬们能不能给鼠鼠提一些意见,万分感谢!!!
牛客78696106...:浪潮可不是开摆,当初我还是开发的时候我组长跟我说他们组有段时间天天1,2点走,早上5点就来,全组肝出来心肌炎,浪潮挣钱省立花可不是说说,当然也看部门,但是浪潮普遍就那dio样,而且你算下时薪就知道不高,没事也是9点半走,不然算你旷工
投递小米集团等公司10个岗位
点赞 评论 收藏
分享
用微笑面对困难:这里面最强的是驾驶证了,可以入职美团大厂,然后直接开启黄马褂人生
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务