BM46-最小的K个数

题目

给定一个长度为 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(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]

---

思路

用JDK内置的优先级队列(基于最小堆)，取小用大 -> 构建最大堆。

---

代码

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));
        //构建一个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;
    }
}