T32,学一招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]也可以        

示例2

输入：

[1],0

返回值：

[]

示例3

输入：

[0,1,2,1,2],3

返回值：

[0,1,1]

方法一：Java真香，偷个懒

 

import java.util.*;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> array=new ArrayList<>(k);
        if(k>input.length){
            return array;
        }
        Arrays.sort(input);
        for(int i=0;i<k;i++){
            array.add(input[i]);
        }
        return array;
    }
}

 

 

 方法二：大根堆

import java.util.*;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> array=new ArrayList<>(k);
        PriorityQueue<Integer> queue=new PriorityQueue<>((o1, o2) -> o2 - o1);
        if(k>input.length||k==0){
            return array;
        }
        for(int i=0;i<k;i++){
            queue.add(input[i]);
        }
        for(int i=k;i<input.length;i++){
            if(queue.peek()>input[i]){
                queue.remove();
                queue.add(input[i]);
            }
        }
        for(int i=0;i<k;i++){
            array.add(queue.remove());
        }
        return array;
        
    }
}

思路：现在大根堆中放入前k个元素，遍历后续元素，如果小于堆顶元素则加入堆中，之后将堆加入数组中，返回数组。注意不能是小根堆，因为堆顶元素有可能是k个最小之中，移除后显然不对，但是同时也给了我们一个提示，如果要输出k个依次减小的数则可以用小根堆。

 附录：

PriorityQueue的使用：java中大小堆的创建_wl1929的博客-CSDN博客_(o1, o2) -> 0，Java 优先队列（PriorityQueue）总结_lee的Csdn的博客-CSDN博客_java priorityqueue