题目:给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。
数据范围:0≤k,n≤10000,数组中每个数的大小0≤val≤1000
要求:空间复杂度 O(n) ,时间复杂度 O(nlogn)
示例1
输入:[4,5,1,6,2,7,3,8],4
返回值:[1,2,3,4]
说明:返回最小的4个数即可,返回[1,3,2,4]也可以
示例2
输入:[1],0
返回值:[]
示例3
输入:[0,1,2,1,2],3
返回值:[0,1,1]
方法一:使用堆排序
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(input.length == 0 || k == 0 || k > input.length)
return list;
//创建小顶堆
for(int i = input.length / 2 - 1; i >= 0; i--) {
adjustHeap(input, i, input.length);
}
//排序
for(int j = input.length - 1; j >= 0; j--) {
int first = input[0];
list.add(first);
if(list.size() == k)
return list;
input[0] = input[j];
adjustHeap(input, 0, j);
}
return list;
}
private void adjustHeap(int[] input, int i, int length) {
int temp = input[i];
for(int l = 2 * i + 1; l < length; l = 2 * l + 1) {
if(l + 1 < length && input[l] > input[l + 1]) {
l++;
}
if(input[l] < temp){
input[i] = input[l];
i = l;
} else {
break;
}
}
input[i] = temp;
}
}
方法二:使用归并排序
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(input.length == 0 || k == 0 || k > input.length)
return list;
int[] temp = new int[input.length];
input = mergeSort(input, 0, input.length - 1, temp);
for(int i = 0; i < k; i ++) {
list.add(input[i]);
}
return list;
}
private int[] mergeSort(int[] input, int left, int right, int[] temp) {
if(left < right) {
int mid = (left + right) / 2;
mergeSort(input, left, mid, temp);
mergeSort(input, mid + 1, right, temp);
merge(input, left, mid, right, temp);
}
return input;
}
//合并
private void merge(int[] input, int left, int mid, int right, int[] temp) {
int i = left;
int j = mid + 1;
int t = 0;
while(i <= mid && j <= right) {
if(input[i] < input[j]) {
temp[t] = input[i];
t++;
i++;
} else {
temp[t] = input[j];
t++;
j++;
}
}
//若左边有剩余
while(i <= mid) {
temp[t] = input[i];
i++;
t++;
}
//若右边有剩余
while(j <= right) {
temp[t] = input[j];
j++;
t++;
}
//将temp数组的元素拷贝到input
t = 0;
int tempLeft = left;
while(tempLeft <= right) {
input[tempLeft] = temp[t];
t++;
tempLeft++;
}
}
}