LeetCode #215 - Kth Largest Element in an Array

最新推荐文章于 2022-02-12 10:28:12 发布

原创最新推荐文章于 2022-02-12 10:28:12 发布 · 103 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

625 篇文章

订阅专栏

本文介绍了一种高效算法，用于从无序数组中找到第K大的元素。通过使用大小为K的优先队列，该算法能在O(nlogk)的时间复杂度内解决问题，优于全排序方法。

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

题目描述：

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.

Example 1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5

Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

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

由于是无序数组，直接排序再求解的时间复杂度为O(nlog n)，但是其实我们只需要求第k大的元素即可，所以我们只需要维持一个程度为k的优先队列，遍历一遍数组并进行插入和删除，时间复杂度为O(nlog k)。

class comp{
    public:
    bool operator()(int a,int b)
    {
        if(a>b) return true;
        else return false;
    }
};

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int,vector<int>,comp> q;
        for(int i=0;i<nums.size();i++)
        {
            if(q.size()<k) q.push(nums[i]);
            else
            {
                q.push(nums[i]);
                q.pop();
            }
        }
        return q.top();
    }
};