题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
方法一,利用工具类排序
import java.util.Arrays;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(k > input.length) return list;
Arrays.sort(input);
for(int i = 0;i < k;i++){
list.add(input[i]);
}
return list;
}
}
方法二,利用插入排序算法先排好序,再取前k个
import java.util.Arrays;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(input == null || input.length == 0 || k > input.length) return list;
//插入排序
int insertVal = 0;
int insertIndex = 0;
for(int i = 1;i < input.length;i++){
insertVal = input[i];
insertIndex = i - 1;
while(insertIndex >= 0 && insertVal < input[insertIndex]){
input[insertIndex + 1] = input[insertIndex];
insertIndex--;
}
//insertIndex + 1 != i说明上面进入了while循环
if(insertIndex + 1 != i){
input[insertIndex + 1] = insertVal;//上面--了,所以+1是insertIndex的正确位置
}
}
//排序后取前k个
for(int i = 0;i < k;i++){
list.add(input[i]);
}
return list;
}
}