题解 | #最小的K个数#

最小的K个数

http://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf

-- coding:utf-8 --

class Solution: def GetLeastNumbers_Solution(self, tinput, k): # write code here

    import heapq
    if  len(tinput) <= k:
        return tinput
    
    hp = []
    for item in tinput:
        if len(hp) < k:
            heapq.heappush(hp,  -item)
        else:
            heapq.heappushpop(hp, -item)
            
    return [-item for item in hp][::-1]
全部评论

相关推荐

喜欢走神的孤勇者练习时长两年半:爱华,信华,等华,黑华
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务