题目:
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.
代码:
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int> pq;
int res = 2147483647;
for(int i = 0; i<nums.size(); i++)
{
pq.push(nums[i]);
if(pq.size() > nums.size() - k)
{
if(res > pq.top())
{
res = pq.top();
}
pq.pop();
}
}
return res;
}
};