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

题目

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

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

传送门 leetcode 215

思路

• 最小堆可以满足需求，offer入队，poll出队，而且不报异常。
• 使用Priorityqueue,遍历一次，当数量超过k时，出列。缺点是占内存。
• 或者使用quickSelect算法。

class Solution {
    public int findKthLargest(int[] nums, int k) {
         if (nums == null || nums.length == 0) return 0;
        PriorityQueue minHeap = new PriorityQueue<>();
        for (int num : nums) {
            minHeap.offer(num);//将数组元素入队
            if (minHeap.size() > k) {
                minHeap.poll();//最小元素出队
            }
        }
        return minHeap.poll();//结果值
    }
}