【LeetCode: 215. 数组中的第K个最大元素 + 堆】-Toy模板网

这篇具有很好参考价值的文章主要介绍了【LeetCode: 215. 数组中的第K个最大元素 + 堆】。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

【LeetCode: 215. 数组中的第K个最大元素 + 堆】,LeetCode每日一题打卡,面试必须掌握的101题,leetcode,算法,java,面试,堆,快速排序

🚀 算法题 🚀

🌲 算法刷题专栏 | 面试必备算法 | 面试高频算法 🍀
🌲 越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
🌲 作者简介：硕风和炜，CSDN-Java领域优质创作者🏆，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享💎💎💎
🌲 恭喜你发现一枚宝藏博主,赶快收入囊中吧🌻
🌲 人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？🎯🎯

🚀 算法题 🚀

【LeetCode: 215. 数组中的第K个最大元素 + 堆】,LeetCode每日一题打卡,面试必须掌握的101题,leetcode,算法,java,面试,堆,快速排序

🚩 题目链接

215. 数组中的第K个最大元素

⛲ 题目描述

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4], k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6], k = 4
输出: 4

提示：

1 <= k <= nums.length <= 105
-104 <= nums[i] <= 104

🌟 求解思路&实现代码&运行结果

⚡ 堆

🥦 求解思路

我们先初始化一个大小为k的小根堆，然后从nums[k+1]开始与堆顶元素比较，如果nums[i] > heap.peek()，则将当前堆顶元素出堆,并将nums[i]入堆，遍历结束后，返回此时堆顶的元素就是最终答案ans。
有了基本的思路，接下来我们就来通过代码来实现一下。

🥦 实现代码

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        for (int i = 0; i < k; i++) {
            heap.add(nums[i]);
        }
        for (int i = k; i < nums.length; i++) {
            if (nums[i] > heap.peek()) {
                heap.poll();
                heap.add(nums[i]);
            }
        }
        return heap.peek();
    }
}