题解 | #连续牛的第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大的。