题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
解题思路:
采用选择排序的思想即可,注意k的范围。
import java.util.ArrayList;
public class Solution {
int len = 0;
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> array = new ArrayList<>();
if (input == null || input.length == 0) {
return array;
}
int len = input.length;
for (int i = 0; i < len; i++) {
int min_number = i;
for (int j = i + 1; j < len; j++) {
if (input[min_number] > input[j]) {
int temp = input[min_number];
input[min_number] = input[j];
input[j] = temp;
}
}
}
if (k <= input.length) {
for (int i = 0; i < k; i++) {
array.add(input[i]);
}
}
return array;
}
}