JZ40 最小的K个数

描述

给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。

思路:

具体做法:
step 1:优先判断k为0或者输入数组长度为0的特殊情况。
step 2:使用sort函数对整个数组排序。
step 3:遍历排序后的数组前k个元素即可获取最小的k个

代码

package mid.JZ40最小的K个数;

import java.util.ArrayList;
import java.util.Arrays;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        //存储结果
        ArrayList<Integer> result = new ArrayList<>();
        //初始化最大元素为1位置的元素
        Arrays.sort(input);
        for (int i = 0; i < k; i++) {
            result.add(input[i]);
        }
        return result;
    }

    public static void main(String[] args) {
        int[] arr = new int[]{4,5,1,6,2,7,3,8};
        System.out.println(new Solution().GetLeastNumbers_Solution(arr, 4));
    }
}