第K大数 小顶堆

寻找第K大

http://www.nowcoder.com/questionTerminal/e016ad9b7f0b45048c58a9f27ba618bf

小顶堆,堆顶为最小元素
空间复杂度O(K)

#include<queue>
class Finder {
public:
    int findKth(vector<int> a, int n, int K) {
        // write code here
        priority_queue<int, vector<int>, greater<int>> Q;
        for(auto x : a){
            Q.push(x);
            if(Q.size() > K){  //保证堆中不超过K个元素
                Q.pop();
            }
        }
        return Q.top();
    }
};
全部评论

相关推荐

06-28 22:48
已编辑
广东金融学院 Java
小浪_Coding:学院本+这俩项目不是buff叠满了嘛
点赞 评论 收藏
分享
06-20 17:42
东华大学 Java
凉风落木楚山秋:要是在2015,你这简历还可以月入十万,可惜现在是2025,已经跟不上版本了
我的简历长这样
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务