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

最新推荐文章于 2025-12-04 23:36:16 发布

原创最新推荐文章于 2025-12-04 23:36:16 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展 #编程

本文介绍如何使用O(n)空间复杂度和O(nlogn)时间复杂度的方法，在给定的可能包含重复值的数组中找到最小的k个数，通过双指针和堆数据结构实现。适用于0到100000的整数范围，且数组元素大小不超过10000。

描述

给定一个长度为 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