题解 | #最小的K个数#
最小的K个数
https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param input int整型一维数组
* @param k int整型
* @return int整型ArrayList
*/
public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
ArrayList<Integer> res = new ArrayList<>();
if(input.length == 0||k == 0)
return res;
int[] sortArr = new int[k];
for(int i = 0;i < k;i++){
sortArr[i]=input[i];
}
//剪枝-保证前k有序
Arrays.sort(sortArr);
for(int i = k;i < input.length;i++){
if(sortArr[k-1] > input[i]){
int temp = input[i];
input[i] = sortArr[k-1];
sortArr[k-1] = temp;
Arrays.sort(sortArr);
}
}
for(int i = 0;i<k;i++){
res.add(sortArr[i]);
}
return res;
}
}

