牛客网 JZ40 最小的K个数 Python代码

描述

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0\le k,n \le 100000≤k,n≤10000，数组中每个数的大小0 \le val \le 10000≤val≤1000

要求：空间复杂度 O(n)O(n) ，时间复杂度 O(nlogn)O(nlogn)

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param input int整型一维数组 
# @param k int整型 
# @return int整型一维数组
#
class Solution:
    def GetLeastNumbers_Solution(self , input: List[int], k: int) -> List[int]:
        # write code here
        if k == 0: return []
        mink = []
        d = 1001
        flag = 0
        for i in range(0, k):
            mink.append(d)
            d += 1
        for i in range(len(input)):
            for j in range(k):
                if input[i] < mink[j]:
                    for m in range(k-1, j, -1):
                        mink[m] = mink[m-1]
                    mink[j] = input[i]
                    break
        return mink