牛客网|最小的k个数

题目信息

给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。如果K>数组的长度，那么返回一个空的数组

示例输入：[4,5,1,6,2,7,3,8],4

示例输出：[1,2,3,4]

题解

将指定数组升序排列，然后取前K个值即可

编码

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

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        Arrays.sort(input);
        if (k > input.length) {
            return new ArrayList<Integer>();
        }
        
        ArrayList<Integer> list = new ArrayList<>();
        for (int i=0; i<k; i++) {
            list.add(input[i]);
        }
        
        return list;
    }
}