题解 | #最小的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;
    }
};
全部评论

相关推荐

不愿透露姓名的神秘牛友
07-08 13:05
点赞 评论 收藏
分享
点赞 评论 收藏
分享
05-29 20:34
门头沟学院 C++
KarlAllen:得做好直接春招的准备。学历差的话,一是面试要求会比学历好的严格不少,二是就算面试通过了也会被排序。总之暑期和秋招对于学历差的就是及其不友好
无实习如何秋招上岸
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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