堆/栈/队列 最小K个数 Java

给定一个长度为 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;
    }
}