手撕算法-数组中的第K个最大元素

描述

分析

使用小根堆，堆元素控制在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();}
}

面试公司