链接
https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/
前言
题目
输入整数数组 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]
提示:
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000
关键
本人思路
- python自带sort()函数
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
arr.sort()
return arr[:k]
思路1
- 快速排序
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
def partition(arr, left, right):
i = left
j = right
base = arr[i]
while i < j:
while arr[j] >= base and i < j:
j-=1
arr[i] = arr[j]
while arr[i] <= base and i < j:
i+=1
arr[j] = arr[i]
arr[i] = base
return i
def quicksort(arr, left, right):
if left < right:
i = partition(arr, left, right)
quicksort(arr, left, i-1)
quicksort(arr, i+1, right)
quicksort(arr, 0, len(arr)-1)
return arr[:k]