leetcode215. Kth Largest Element in an Array

最新推荐文章于 2025-08-22 19:21:44 发布

原创最新推荐文章于 2025-08-22 19:21:44 发布 · 216 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #heap #array

leetcode 专栏收录该内容

187 篇文章

订阅专栏

本文介绍了一种使用优先队列解决寻找未排序数组中第K大元素的方法。通过维护一个大小为K的优先队列，可以高效地找到目标元素。

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

215. Kth Largest Element in an Array

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note:
You may assume k is always valid, 1 ≤ k ≤ array’s length.

解法

维护长度为k的优先队列

public class Solution {
    public int findKthLargest(int[] nums, int k) {
        if (nums == null || nums.length == 0 || k > nums.length) {
            return -1;
        }

        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int num : nums) {
            pq.offer(num);
            if (pq.size() > k) {
                pq.poll();
            }
        }

        return pq.peek();
    }
}