【LeetCode】 215. 数组中的第K个最大元素大顶堆

最新推荐文章于 2024-04-07 02:17:07 发布

Geek-Banana

最新推荐文章于 2024-04-07 02:17:07 发布

阅读量406

点赞数

分类专栏： LeetCode刷题记录与总结文章标签： leetcode 堆 java

本文链接：https://blog.csdn.net/qq1515312832/article/details/104516537

版权

LeetCode刷题记录与总结专栏收录该内容

62 篇文章

订阅专栏

本文介绍了一种高效查找数组中第K大元素的方法，通过使用最大堆的数据结构，避免了传统排序方法的高时间复杂度。文章详细解释了如何构建和维护最大堆，以确保堆中始终保留K个最大的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

题目传送门：传送门（点击此处）
在这里插入图片描述

题解

思路

这道题目是有难度的，如果使用先排序的方法，就有一点墨迹了，所以我们这道题目借助堆的数据结构，实现最大堆就可以了

代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        // init heap 'the smallest element first'
        PriorityQueue<Integer> heap = new PriorityQueue<Integer>((n1, n2) -> n1 - n2);
        // keep k largest elements in the heap
        for (int n : nums) {
            heap.add(n);
            if (heap.size() > k)
                heap.poll();
        }
        // output
        return heap.poll();
    }
}