輸入整數數組 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]鏈接：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcofclass Solution:def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:if k >= len(arr):return arrdef quick_sort(l, r):i, j = l, rwhile i < j:while i < j and arr[j] >= arr[l]:j -=1while i < j and arr[i] <= arr[l]:i +=1arr[i], arr[j] = arr[j], arr[i]arr[l], arr[i] = arr[i], arr[l]if k < i:return quick_sort(l, i-1)if k > i:return quick_sort(i + 1, r)return arr[:k]return quick_sort(0, len(arr)-1)

總結

以上是生活随笔為你收集整理的Leetcode 剑指 Offer 40. 最小的k个数 (每日一题 20210825)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。