面試題40.最小的k個數

面試題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]

限制：

0 <= k <= arr.length <= 1000
0 <= arr[i] <= 1000

題解

排序數組，返回前k個數，剛開始不會用Arrays.copyOf(arr,k)方法，藉助了棧（參考註釋部分），兩次提交結果代碼如下：

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        Arrays.sort(arr);
        /*int[] arr2 = new int[k];
        for(int i = 0;i<k;i++){
            arr2[i] = arr[i];
        }
        return arr2;
        */
    return Arrays.copyOf(arr,k);
    }
}

提交結果