题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
解法一:
对整个数组进行排序,排序后直接取前k个数。
代码如下:
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<>();
if (input == null || input.length < k){
return result;
}
Arrays.sort(input);
for (int i = 0; i < k; i++) {
result.add(input[i]);
}
return result;
}
利用快速排序的基础算法来实现,当我们取数组中第k-1的数,假如比它小的数都在它的左边,则前k个数就是数组中最小的k个数
代码如下:
public static void swap(int[] input, int x, int y){
int t = input[x];
input[x] = input[y];
input[y] = t;
}
public static int partition(int[] input, int start, int end){
if (input == null ||
input.length <= 0 ||
start < 0 ||
end >= input.length){
return -1;
}
int index = new java.util.Random()
.nextInt(end - start) + start;
swap(input, index, end);
int small = start - 1;
//遍历数组段
for (index = start; index < end; index++) {
if (input[index] < input[end]){
small ++;
if (small != index){
swap(input, index, small);
}
}
}
++ small;
swap(input, small, end);
return small;
}
public static ArrayList<Integer> GetLeastNumbers_Solution2(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<>();
if (input == null || input.length < k || k <= 0){
return result;
}
if (k != input.length){
int start = 0;
int end = input.length - 1;
int index = partition(input, start, end);
while (index != k - 1){
if (index > k - 1){
end = index - 1;
index = partition(input, start, end);
}else{
start = index + 1;
index = partition(input, start, end);
}
}
}
Arrays.sort(input, 0, k);
for (int i = 0; i < k; i++) {
result.add(input[i]);
}
return result;
}
解法三:
如果有海量的数据,我们创建一个大小为k的辅助空间,同时标志最大值,当我们从头到尾遍历整个输入数组,当辅助空间未满,则直接放入辅助空间,如果满了而且当前遍历值小于辅助空间最大值,则替换辅助空间最大值,同时查找最大值。
代码如下:
public static ArrayList<Integer> GetLeastNumbers_Solution3(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<>();
if (input == null || input.length < k || k <= 0){
return result;
}
int max = 0;
if (k != input.length){
for (int i = 0; i < input.length; i++) {
if (result.size() < k){
result.add(input[i]);
if (result.get(max) < input[i]){
max = result.size() - 1;
}
}else if (result.get(max) > input[i]){
result.set(max, input[i]);
max = 0;
for (int j = 1; j < result.size(); j++) {
if (result.get(max) < result.get(j)){
max = j;
}
}
}
}
}else{
for (int i = 0; i < k; i++) {
result.add(input[i]);
}
}
Collections.sort(result);
return result;
}