利用python自带的小根堆 LeetCode

最新推荐文章于 2024-02-02 17:57:23 发布

weixin_40248634

最新推荐文章于 2024-02-02 17:57:23 发布

阅读量754

点赞数

分类专栏： coding 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_40248634/article/details/128047012

版权

coding 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

求第小的k个数

python 的 heapq 是小根堆，所以需要转化为大根堆会方便处理，即将数组取为负数。

class Solution:
    def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
        if k==0:
            return []

        hp=[-x for x in arr[:k]]
        heapq.heapify(hp)
        for i in arr[k:]:
            if -hp[0]>i:
                heapq.heappop(hp)
                heapq.heappush(hp,-i)
        
        return [-x for x in hp]