LeetCode215：数组中的第k个最大元素

最新推荐文章于 2024-11-12 21:59:57 发布

#小学生

最新推荐文章于 2024-11-12 21:59:57 发布

阅读量143

点赞数

分类专栏：算法/LeetCode 文章标签：算法排序算法数据结构 java intellij-idea

本文链接：https://blog.csdn.net/weixin_46426906/article/details/127475314

版权

算法/LeetCode 专栏收录该内容

101 篇文章 0 订阅

订阅专栏

该博客介绍了如何在Java中利用优先队列(PriorityQueue)实现寻找整数数组中第k个最大元素的算法。通过创建一个大小为k的小顶堆，在遍历数组过程中不断调整堆，最终堆顶元素即为第k大的数。这种方法避免了排序，达到了O(n)的时间复杂度。

摘要由CSDN通过智能技术生成

要求

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。
请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
在这里插入图片描述

题目解析

数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。
这句话的意思是，如果数组当中有值相同的元素，他们不能够合并成一个元素；并且排序是从大到小

思路

第一种：你可以直接暴力破解：使用sort直接排序获取目标值
第二章：优先队列
数组排序后的第 k 个最大的元素：换句话说就是从右往左数第k个元素，即数组长度len - k；优先队列可以用一个 for 循环解决。在 for 循环里面判断小顶堆里面的 size() 是否大于 k 个数，是的话就 poll() 出去；整个 for 循环结束之后剩下来的就是 k 个数的小顶堆。堆顶即第 k 大的数。

public int findKthLargest(int[] nums, int k) {
    // 使用一个含有 k 个元素的最小堆，PriorityQueue 底层是动态数组，为了防止数组扩容产生消耗，可以先指定数组的长度
    PriorityQueue<Integer> heap = new PriorityQueue<>();
    //遍历整数数组，添加到动态数组中
    for (int num : nums) {
        heap.add(num);
        if (heap.size() > k){
            heap.poll();
        }
    }
    return heap.peek();
}