题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
示例1
输入:
[4,5,1,6,2,7,3,8],4
返回值:
[1,2,3,4]
牛客编译通过代码
1、代码
牛客的python环境为python 2.7
# -*- coding:utf-8 -*-
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
tinput = sorted(list(tinput))
newlist = []
if len(tinput) >= k:
newlist = tinput[0:k]
return newlist
else:
return newlist
# write code here
2、时长和内存
3、算法思想
找出一组数中最小(最大)的K个数,我们可以先排序,在切片。
(1)排序函数:sorted()函数
sorted() 函数对所有可迭代的对象进行排序操作,内建函数 sorted 方法返回的是一个新的 list,而不是在原来的基础上进行的操作,原可迭代对象不变。
sorted 语法:
sorted(iterable, cmp=None, key=None, reverse=False)
参数说明:
1、iterable – 可迭代对象。
2、cmp – 比较的函数,这个具有两个参数,参数的值都是从可迭代对象中取出,此函数必须遵守的规则为,大于则返回1,小于则返回-1,等于则返回0。
3、key – 主要是用来进行比较的元素,只有一个参数,具体的函数的参数就是取自于可迭代对象中,指定可迭代对象中的一个元素来进行排序。
4、reverse – 排序规则,reverse = True 降序 , reverse = False 升序(默认)。
(2)切片:[起始index:终止index:步长]
切片时 前闭后开