面试题40. 最小的k个数
难度:简单
题目:
输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
示例 1:
输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]
示例 2:
输入:arr = [0,1,2,1], k = 1
输出:[0]
方法一:排序
思路和算法
对原数组从小到大排序后取出前 kk 个数即可。
复杂度分析
时间复杂度:O(n\log n)O(nlogn),其中 nn 是数组 arr 的长度。算法的时间复杂度即排序的时间复杂度。
空间复杂度:O(\log n)O(logn),排序所需额外的空间复杂度为 O(\log n)O(logn)。
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
arr.sort()
return arr[:k]