描述
分析
使用小根堆,堆元素控制在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();}
}