输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
方法一:排序后返回前K个值
# -*- coding:utf-8 -*-
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
res=[]
for i in tinput:
res.append(i)
res.sort()
#注意判断
if k > len(res):
return []
return res[:k]
方法二:python的内置模块heapq,堆的push和pop
import heapq
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
heaps = []
ret = []
for num in tinput:
heapq.heappush(heaps,num)
注意判断
if k>len(heaps):
return []
for i in range(k):
ret.append(heapq.heappop(heaps))
return ret