【每日一题】【优先队列、迭代器、lambda表达式】2022年1月15日-NC119 最小的K个数

描述

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0\le k,n \le 100000≤k,n≤10000，数组中每个数的大小0 \le val \le 10000≤val≤1000

要求：空间复杂度 O(n)O(n) ，时间复杂度 O(nlogn)O(nlogn)

方法：优先队列、迭代器、lambda表达式

import java.util.*;

public class Solution {
    //堆的使用、lambda表达式的使用、迭代器的使用
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res = new ArrayList<>();
        if(input.length == 0 || k == 0) {
            return res;
        }
        //怎么判断是大根堆还是小根堆，目的是大根堆
        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b - a);
        for(int i = 0; i < input.length; i++) {
            if(queue.size() < k) {
                queue.add(input[i]);
            } else if(queue.peek() > input[i]) {
                queue.poll();
                queue.add(input[i]);
            }
        }
        Iterator<Integer> iterator = queue.iterator();
        while(iterator.hasNext()) {
            res.add(iterator.next());
        }
        return res;
    }
}

要保证堆的大小不能超过K，然后遍历数组，因为是最大堆，也就是堆顶元素是堆中最大的，如果遍历的元素小于堆顶元素，就把堆顶元素给移除，然后再把当前遍历的元素加入到堆中，最后在把堆中元素转化为数组即可。

怎么判断是大根堆：(a,b) -> b - a，小-大