题解 | #最小的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]