215. 数组中的第K个最大元素

题目描述

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

示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

解法

class Solution {
    public int findKthLargest(final int[] nums, final int k) {
        // 创建一个大小为K的最小堆，堆顶就是第K大的数字

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

        //创建堆
        for(int i=0; i<k; i++){
            queue.offer(nums[i]);
        }

        //遍历后续的元素
        for(int i=k; i<nums.length; i++){
            //如果元素大于堆顶，则将堆顶弹出，将元素入堆
            if(nums[i] > queue.peek()){
                queue.poll();
                queue.offer(nums[i]);
            }
        }

        //返回堆顶的元素就是第k大的
        return queue.peek();
    }
}