1.题目
2.解法(使用插入排序)
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<Integer>();
if(k<= 0 || k > input.length)return result;
//初次排序,完成k个元素的排序
for (int i = 1; i < k; i++) {
int j = i;
int tmp = input[i];
while(j > 0 && input[j-1] > tmp) {
input[j] = input[j-1];
--j;
}
input[j] = tmp;
}
//遍历后面的元素 进行k个元素的更新和替换
for (int i = k; i < input.length; i++) {
if (input[i] < input[k-1]) {
int tmp = input[i];
int j = k - 1;
while (j > 0 && input[j-1] > tmp) {
input[j] = input[j-1];
--j;
}
input[j] = tmp;
}
}
//把前k个元素返回
for (int i = 0; i < k; i++)
result.add(input[i]);
return result;
}
}
先排好前k个数,然后再1个1个的比
== 最坏情况下时间复杂度为O(n^2),空间复杂度为O(k)==