题解 | #最小的K个数#

最小的K个数

http://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf

topK问题,用一个大小为K的堆实现。

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> result{};
        if(k==0) return result;
        
        priority_queue<int,vector<int>,less<int>> max_heap{};
        for(int i=0;i<k;i++){
            max_heap.push(input[i]);
        }
        for(int i=k;i<input.size();++i){
            if(input[i]<max_heap.top()){
                max_heap.pop();
                max_heap.push(input[i]);
            }
        }
        
        while(!max_heap.empty()){
            result.push_back(max_heap.top());
            max_heap.pop();
        }
        return result;
    }
};
全部评论

相关推荐

不愿透露姓名的神秘牛友
昨天 14:08
点赞 评论 收藏
分享
05-26 16:13
门头沟学院 C++
牢大肘击Java:海投就完事了bro,就当刷视频了
点赞 评论 收藏
分享
07-07 11:33
江南大学 Java
已经在暑假实习了&nbsp;,没有明确说有hc,纠结实习到八月份会不会有点影响秋招毕竟感觉今年好多提前批
程序员小白条:92的话准备提前批,其他没必要,没面试机会的,而且你要准备充分,尤其八股和算法题
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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