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

最新推荐文章于 2024-07-25 11:27:31 发布

迷思特王

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量401

点赞数 4

分类专栏：手撕算法能力提升计划文章标签：算法 java 数据结构

本文链接：https://blog.csdn.net/weixin_43155866/article/details/137058317

版权

手撕算法能力提升计划专栏收录该内容

32 篇文章 1 订阅

订阅专栏

描述

分析

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