题解 | #最小的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; } }