题目描述
在未排序的数组中找到第 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 ≤ 数组的长度。
思路
这题利用了 priority_queue 优先队列,默认声明 priority_queue<int> q 的时候 q 是大根堆,即堆中元素从大到小排列(堆顶永远是堆中最大元素),按代码中定义 q 是小根堆,即堆顶元素永远是该堆中的最小元素。
本题中,先把 k 个元素加入堆中,这时堆顶元素应为前 k 个元素中第 k 大的,然后遍历数组剩余元素,若遍历的元素大于堆顶元素,即替换堆顶元素,这样遍历结束后堆顶元素即为所求。
代码
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
priority_queue<int, vector<int>, greater<int>> q;
for(int i=0;i<nums.size();i++){
if(i < k){
q.push(nums[i]);
}else if(nums[i] > q.top()){
q.pop();
q.push(nums[i]);
}
}
return q.top();
}
};