题解 | #最小的K个数#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param input int整型一维数组 
# @param k int整型 
# @return int整型一维数组
#
# 每次将保存的四个值中的最大值与当前值比较
# 若当前值更小，则更新答案
class Solution:
    def GetLeastNumbers_Solution(self , input: List[int], k: int) -> List[int]:
        if k == 0:
            return None
        elif k == 1:
            return [min(input)]
        
        ans = [1000 for _ in range(k)]
        for i in input:
            t = max(ans)
            if i <= t:
                ans[ans.index(t)] = i
        
        return ans