最大/最小de K个数/第K个数

Posted 2021-03-08 Jun10ng

题目

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

思路

堆排序

收获

用优先队列实现最大最小堆
注意下列代码中优先队列的创建声明，需要自定义比较函数

代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> h = new PriorityQueue<>((n1,n2)->n1-n2);
        for(int i:nums){
            h.add(i);
            if(h.size()>k){
                h.poll();
            }
        }
        return h.poll();
    }
}