大顶堆或者小顶堆PriorityQueue解决TopK和k个最值问题

1. PriorityQueue用法

容量k,默认为小根堆,即堆顶的元素为堆的最小值。

PriorityQueue<Integer> queue = new PriorityQueue<>(K);

通过lambdas表达式设置排序方式,这里设置大顶堆,即堆顶的元素为堆的最大值。

PriorityQueue<Integer> queue = new PriorityQueue<>(K, (o1, o2) -> o2 - o1);

2. 第k个值的问题

如果是第k大,则可以用大顶堆排序,依次弹出k-1个数,则堆顶元素就是第k大。
如果是第k小,则可以用小顶堆排序,依次弹出k-1个数,则堆顶元素就是第k小。

牛客第BM47 题:寻找第K大

public class Solution {
    public int findKth (int[] a, int n, int K) {
        if (n <= 0) {
            return -1;
        }
        //第k大,按照从大到小排序,取第k个就可以。
        PriorityQueue<Integer> queue = new PriorityQueue<>(K, (o1, o2) -> o2 - o1);
        for (int i = 0; i < n; i++) {
            queue.add(a[i]);
        }
        for (int i = 0; i < K - 1; i++) {
            queue.poll();
        }
        return queue.poll();
    }
}

3. 最大或者最小的k个值

如果是k大,则可以用大顶堆排序,依次弹出k个数。
如果是k小,则可以用小顶堆排序,依次弹出k个数。

牛客BM46 最小的K个数:

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        ArrayList<Integer> res = new ArrayList<>();
        if (input.length == 0 || k == 0) {
            return res;
        }
        //最小的k个数,按照从小到大排序,取前k个就可以。
        PriorityQueue<Integer> queue = new PriorityQueue<>(k);
        for (int i = 0; i < input.length; i++) {
            queue.add(input[i]);
        }
        for (int i = 0; i < k; i++) {
            res.add(queue.poll());
        }
        return res;
    }
}

你可能感兴趣的:(算法,java,数据结构,PriorityQueue)