數組中的第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

說明:

你可以假設 k 總是有效的，且 1 ≤ k ≤ 數組的長度。

來源：力扣
鏈接：數組中第k個最大元素

思路:

1. 利用 priority_queue優先級隊列, 建立k個數據小堆,

2. 將nums中剩餘數據逐一與堆頂數據比較, 大於堆頂數據就刪除原數據, 插入新數據

3. 最終堆頂爲第k大元素
   代碼示例:

   class Solution {
   public:
   int findKthLargest(vector& nums, int k) {
   // return pq.top();
   priority_queue<int, vector, greater> pq;
   size_t i = 0;
   //建立k個數據的小堆
   for (; i < k; ++i)
   {
   pq.push(nums[i]);
   }

        //將nums中剩餘的數據與優先級隊列(堆頂)比較,大於其堆頂數據,就刪除堆頂數據,再插入新數據
        for (; i < nums.size(); ++i)
        {
            if (nums[i] > pq.top())
            {
                pq.pop();
                pq.push(nums[i]);
            }
        }
        return pq.top();//循環結束後,堆頂即爲第k大數據
    }
   

   };