最小的K个数

Posted 2021-02-13 IT嘟嘟

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

 

# -*- coding:utf-8 -*-
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        #如果输入的n不存在则输出[]空list,如果K大于输入的长度也是同样输出为空的
        #对输入的数据进行排序，去0：k的数即为所取得的最小的k个数
        #
        n = len(tinput)
        if n == 0 or k > n:
            return []
        else:
            return sorted(tinput)[:k]