LeetCode-算法-最小的k個數

力扣題目地址: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

解決思路
首先用冒泡排序,將arr變爲有序數組。然後取最前面k位的值。

代碼實現:

	/**
     * 最小的k個數
     * https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/
     * @param arr
     * @param k
     * @author Geyuxuan 2020-03-21 00:34:42
     * @return int[]
     */
    public int[] getLeastNumbers(int[] arr, int k) {
        for(int i = 0; i < arr.length; i++) {
            for(int j = 0; j < arr.length-i-1; j++) {
                if(arr[j]>arr[j+1]){
                    int x = arr[j];
                    arr[j] = arr[j+1];
                    arr[j+1] = x;
                }
            }
        }
        int[] result = new int[k];
        for(int i = 0; i < result.length; i++) {
            result[i] = arr[i];
        }
        return result;
    }

不忘初心,砥礪前行。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章