Arrays.sort()对数组进行排序,然后要多少取多少

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

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