描述

分析

使用小根堆，堆元素控制在k个，遍历数组构建堆，最后堆顶就是第K个最大的元素。

代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        // 小根堆
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        for (int i = 0; i < nums.length; i++) {
            if (queue.size() < k) {
                // 堆元素数小于k，直接放入
                queue.offer(nums[i]);
            } else {
                if (nums[i] >= queue.peek()) {
                    // 当前元素大于堆顶元素，弹出堆顶，放入新的
                    queue.poll();
                    queue.offer(nums[i]);
                }
            }
        }
        return queue.peek();
    }
}

面试公司