题目地址

  • 😣 第一次练习 2020å¹´3月20日 思路还是比较清晰,这个题的解法应该很多,但是为了更加熟悉优先队列,就采用优先队列的解法来解 🐂🍻
  • 💩 第二次练习

# 优先队列,建立最小堆

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] ret = new int[k];

        if (arr.length == 0 || k == 0)
            return ret;

        PriorityQueue<Integer> pq = new PriorityQueue<>((a, b) -> b - a);
        for (int i : arr) {
            if (pq.size() < k)
                pq.add(i);
            else if (i < pq.peek()) {
                pq.remove();
                pq.add(i);
            }
        }

        int index = 0;
        while(!pq.isEmpty())
            ret[index ++] = pq.remove();

        return ret;
    }
}

# 快排变形

# 易错点

  • 注意判断条件为空的情况
最后编辑时间: 7/14/2020, 9:21:47 AM