堆排序,利用优先队列priority_que来做

输入n个整数,输出其中最小的k个

http://www.nowcoder.com/practice/69ef2267aafd4d52b250a272fd27052c

#include <iostream>
#include <vector>
#include <set>
#include<algorithm>
#include<string>
#include<queue>
using namespace std;

int main() {
  int n;
    while(cin>>n){
        int k;
        cin>>k;
        priority_queue<int>que;//默认是大顶堆
        for(int i =0;i<n;i++){
            int x;
            cin>>x;
            if(que.size()<k)que.push(x);
            else {
                if(x<que.top()){
                    que.pop();
                    que.push(x);
                }
            }
        }
        vector<int>vet;
        while(que.size()>0){
            vet.push_back(que.top());
            que.pop();
        }
        reverse(vet.begin(), vet.end());
        for(auto c:vet){
            cout<<c<<" ";
        }
        cout<<endl;
    }
}
全部评论

相关推荐

昨天 11:41
门头沟学院 Java
点赞 评论 收藏
分享
想按时下班的大菠萝在...:隔壁学校的,加油多投, 实在不好找可以下个学期开学找,把算法八股准备好,项目有空再换换
投了多少份简历才上岸
点赞 评论 收藏
分享
05-29 22:11
门头沟学院 Java
Elastic90:抛开学历造假不谈,这公司的招聘需求也挺怪的,Java开发还要求你有图文识别、移动端开发和c++的经验,有点逆天了。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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