给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。
数据范围:0≤k,n≤100000≤k,n≤10000,数组中每个数的大小0≤val≤10000≤val≤1000
要求:空间复杂度 O(n)O(n) ,时间复杂度 O(nlogk)O(nlogk)
示例1
输入:[4,5,1,6,2,7,3,8],4
返回值:[1,2,3,4] 说明:返回最小的4个数即可,返回[1,3,2,4]也可以
思路:取k个数为最小数,用堆排序方法求k个数的最大值。遍历数组中剩下的数,依次与最大值比较,如果小于最大值则进行交换,交换后调整大顶堆,得到新的最大值。
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param input int整型一维数组
* @param k int整型
* @return int整型ArrayList
*/
public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
// write code here
ArrayList<Integer> a=new ArrayList<Integer>();
if (k==0||input.length==0){
return a;
}
int num[] =new int[k];
for(int i=0;i<k;i++){
num[i]=input[i];
}
sort(num,k);
for(int j=k;j<input.length;j++){
if(input[j]<num[0]){
num[0]=input[j];
adjustHeap(num,0,k);
}
}
for(int i=k-1;i>=0;i--){
a.add(num[i]);
}
return a;
}
public void sort(int[] num, int k ){
for (int i=k/2-1;i>=0;i--){
adjustHeap(num,i,k);
}
}
public void adjustHeap(int[] arr,int i,int l){//l长度
int temp=arr[i];
for(int k=2*i+1;k<arr.length;k=k*2+1){
if(k+1<arr.length&&arr[k]<arr[k+1]){
k++;
}
if(arr[k]>temp){
arr[i]=arr[k];
i=k;
}else{
break;
}
}
arr[i]=temp;
}
}
法2!使用java优先队列接口PriorityQueue()
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> res = new ArrayList<Integer>();
//排除特殊情况
if(k == 0 || input.length == 0)
return res;
//大根堆
PriorityQueue<Integer> q = new PriorityQueue<>((o1, o2)->o2.compareTo(o1));
//自定义比较器 (o1, o2) -> o2.compareTo(o1):这是一个基于 Lambda 表达式的比较器,它实
//现了降序排列。
//构建一个k个大小的堆
for(int i = 0; i < k; i++)
q.offer(input[i]);
for(int i = k; i < input.length; i++){
//较小元素入堆
if(q.peek() > input[i]){
q.poll();
q.offer(input[i]);
}
}
//堆中元素取出入数组
for(int i = 0; i < k; i++)
res.add(q.poll());
return res;
}
}