关注
#include <iostream>
(30316)#include <queue>
#include <vector>
using namespace std;
class Solution {
public:
vector<int> findKSmallest(vector<int>& nums, int k) {
// 定义一个大顶堆(使用priority_queue,默认是大顶堆)
priority_queue<int> max_heap;
for (int num : nums) {
if (max_heap.size() < k) {
max_heap.push(num);
}
else if (num < max_heap.top()) {
max_heap.pop();
max_heap.push(num);
}
}
vector<int> result;
while (!max_heap.empty()) {
result.push_back(max_heap.top());
max_heap.pop();
}
return result;
}
};
int main() {
vector<int> nums = {4, 2, 7, 1, 9, 3, 5};
int k = 3;
Solution solution;
vector<int> result = solution.findKSmallest(nums, k);
for (int num : result) {
cout << num << " ";
}
cout << endl;
return 0;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 实习要如何选择和准备? #
8567次浏览 228人参与
# 牛友们的论文几号送审 #
18865次浏览 523人参与
# 携程求职进展汇总 #
202791次浏览 1702人参与
# 美团求职进展汇总 #
1622756次浏览 14620人参与
# 满分简历要如何准备? #
14842次浏览 275人参与
# 字节求职进展汇总 #
707369次浏览 7177人参与
# 讲讲我的真实离职原因 #
32964次浏览 621人参与
# 牛友打假中心 #
37429次浏览 2311人参与
# 我想去国央企的原因 #
53048次浏览 322人参与
# 正在实习的你,几点下班 #
98855次浏览 713人参与
# 面试等了一周没回复,还有戏吗 #
97747次浏览 895人参与
# 面试被问第一学历差时该怎么回答 #
103980次浏览 658人参与
# 金蝶求职进展汇总 #
39695次浏览 227人参与
# 稳定和高薪机械人更看重哪个? #
421979次浏览 5293人参与
# 工作两年想退休了 #
98424次浏览 1031人参与
# 德州仪器求职进展汇总 #
2996次浏览 136人参与
# 滴滴求职进展汇总 #
129522次浏览 1326人参与
# 机械人,你的第一份感谢信是谁给的 #
21013次浏览 273人参与
# 通信和硬件还有转码的必要吗 #
46641次浏览 478人参与
# 读研or工作,哪个性价比更高? #
39901次浏览 590人参与
# 小米提前批笔试难吗 #
28751次浏览 317人参与