题目描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
思路:
最直观的思路就是采用一个长度为k的数组nums,保证数组里边的数有序,每访问输入数组input的一个节点,将其与nums数组的值最大的元素即nums[nums.length - 1]进行比较,如果input节点更小,则替换。
本题也要注意各种异常的排除,包括:
输入数组为空,输入的k值为0,k值大于数组长度。
参考代码:
import java.util.ArrayList;
import java.util.Arrays;
public class MinKNumbers {
public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
if (input == null || input.length < k || k == 0) {
return new ArrayList<Integer>();
}
int[] num = new int[k];
for (int p = 0; p < k; p++) {
num[p] = input[p];
}
Arrays.sort(num);
System.out.println(Arrays.toString(num));
for (int i = k - 1; i < input.length; i++) {
if (input[i] < num[k - 1]) {
num[k - 1] = input[i];
Arrays.sort(num);
}
}
ArrayList<Integer> aList = new ArrayList<>();
for (int j = 0; j < k; j++) {
aList.add(num[j]);
}
return aList;
}
public static void main(String[] args) {
int[] input = { 4, 5, 1, 6, 2, 7, 3, 8 };
int k = 8;
ArrayList<Integer> arrayList = GetLeastNumbers_Solution(input, k);
for(int i: arrayList) {
System.out.print(i+" ");
}
}
}