知识点:数组、高级算法
题目描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3, 8 这8个数字,则最小的4个数字是1,2,3,4。
思路:
使用sorted()
函数将n个整数排好序,然后返回前 k 个数字。
解法:
# -*- coding:utf-8 -*-
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
if not tinput or k>len(tinput):
return []
return sorted(tinput)[0:k]
代码测试:
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
if not tinput or k>len(tinput):
return []
return sorted(tinput)[0:k]
if __name__=='__main__':
s=Solution()
rs=s.GetLeastNumbers_Solution([4,5,1,6,2,7,3,8],4)
print(rs)
运行结果如下: