Leetcode 面試題40.最小的k個數

Leetcode 面試題40.最小的k個數

1 題目描述（Leetcode題目鏈接）

  輸入整數數組 arr ，找出其中最小的 k 個數。例如，輸入4、5、1、6、2、7、3、8這8個數字，則最小的4個數字是1、2、3、4。

輸入：arr = [3,2,1], k = 2
輸出：[1,2] 或者 [2,1]

輸入：arr = [0,1,2,1], k = 1
輸出：[0]

限制：

• 0 <= k <= arr.length <= 10000
• 0 <= arr[i] <= 10000

2 題解

  排序類的做法都可以。

class Solution:
    def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
        arr.sort()
        return arr[:k]

更多關於堆的內容

class Solution:
    def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
        return heapq.nsmallest(k, arr)