题解 | #连续牛的第k体重#
连续牛的第k体重
https://www.nowcoder.com/practice/2db4e8f1a0d742e993d7cf8ead90221a
#include <functional>
#include <queue>
#include <vector>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型vector
* @param k int整型
* @return int整型vector
*/
vector<int> findKthLargest(vector<int>& nums, int k) {
// write code here
for(auto num : nums) {
heap.push(num);
if(heap.size() > k) heap.pop();
if (heap.size() == k) ans.push_back(heap.top());
}
return ans;
}
priority_queue<int, vector<int>, greater<>> heap = priority_queue<int, vector<int>, greater<>>();
vector<int> ans = vector<int>();
};
动态维护第k大,只考虑插入的情景。
小顶堆维护最大的k个,堆顶就是第k大的。
查看6道真题和解析