描述
给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。
数据范围:0\le k,n \le 100000≤k,n≤10000,数组中每个数的大小0 \le val \le 10000≤val≤1000
要求:空间复杂度 O(n)O(n) ,时间复杂度 O(nlogk)O(nlogk)
我的解法:
采用归并排序的思想,将数组分解成若干数组,再排序,取出前K个数
也可使用快排,但不稳定,归并排序比较稳定,采用空间换时间的办法
//分治将input排序,取前K个
ArrayList<Integer> temp = new ArrayList<>();
public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
if(input.length<=0||k<=0)return temp;
a(input, 0, input.length - 1,new int[input.length]);
for(int i=0;i<k;i++){
temp.add(input[i]);
}
return temp;
}
public void a(int[] nums, int left, int right, int[] temp) {
if (left == right) { //当拆分到数组当中只要一个值的时候,结束递归
return;
}
int mid = (left + right) / 2; //找到下次要拆分的中间值
a(nums, left, mid, temp); //记录树左边的
a(nums, mid + 1, right, temp); //记录树右边的
//合并两个区间
for (int i = left; i <= right; i++) {
temp[i] = nums[i];
}
int i = left; //给辅助数组里面的值标点
int j = mid + 1;
for (int k = left; k <= right ; k++) {//k 就为当前要插入的位置
if (i == mid + 1) {//此时左边的已经走完了,继续覆盖右边
nums[k] = temp[j];
j++;
} else if (j == right + 1) {//此时右边的已经走完了,继续覆盖左边
nums[k] = temp[i];
i++;
} else if (temp[i] <= temp[j]) {
nums[k] = temp[i];
i++;
} else {
nums[k] = temp[j];
j++;
}
}
}