【leetcode】215. 數組中的第K個最大元素

215. 數組中的第K個最大元素

暴力法

排序，然後找出第n-k個最大的數。

int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        return nums[nums.size()-k];
    }

時間複雜度$O（nlog_2n）$

小頂堆

int findKthLargest(vector<int>& nums, int k)
{
    priority_queue<int, vector<int>, greater<int>> pq;
    for (auto& n : nums)
    {
        if (pq.size() >= k && pq.top() >= n) continue;
        pq.push(n);
        if (pq.size() > k)
        {
            pq.pop();
        }
    }
    return pq.top();
}