给定整数数组 nums
和整数 k
,请返回数组中第 k
个最大的元素。
请注意,你需要找的是数组排序后的第 k
个最大的元素,而不是第 k
个不同的元素。
你必须设计并实现时间复杂度为 O(n)
的算法解决此问题。
示例 1:
输入: [3,2,1,5,6,4],
k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6],
k = 4
输出: 4
思路一:优先队列
c++解法
class Solution {
public:int findKthLargest(vector<int>& nums, int k) {if (k <= 0 || k > nums.size() || nums.empty()) {return 0;}priority_queue<int, vector<int>, greater<int>> que;for (int i = 0; i < nums.size(); i++) {que.push(nums[i]);if (que.size() == k+1) {que.pop();}}return que.top();}
};
分析:
本题要求第k大的元素,利用优先队列可以不断维护排序好的数组,最后直接返回第k大的元素
总结:
本题考察对优先队列的利用,这里采用编写好的优先队列方法,将数组内数直接放入队列中进行排序