Python3 JZ40 最小的K个数（剑指offer）

题目描述

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

Python3 代码

# -*- coding:utf-8 -*-
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        res = []#返回结果
        if k<=len(tinput):#当k值小于或等于tinput长度时
            for i in range(k):#循环k次
                res.append(min(tinput))#每次取最小
                tinput.remove(min(tinput))#移除当前最小
            return res
        return []#当k值大于tinput长度时,返回空