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

全部评论

相关推荐

xwqlikepsl:感觉很厉害啊,慢慢找
点赞 评论 收藏
分享
JamesGosling1:同一个公司的实习为什么写三次,就算是不同的小组的话,直接写一段要好点吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务