关注
#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;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
04-24 18:58
门头沟学院 Java 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 找工作,行业重要还是岗位重要? #
6579次浏览 84人参与
# 盲审过后你想做什么? #
12245次浏览 108人参与
# 五一之后,实习真的很难找吗? #
43900次浏览 311人参与
# 领导秒批的请假话术 #
9478次浏览 72人参与
# 安克创新求职进展汇总 #
32463次浏览 413人参与
# 如果不工作真的会快乐吗 #
100808次浏览 862人参与
# 每人推荐一个小而美的高薪公司 #
72812次浏览 1357人参与
# 京东工作体验 #
12941次浏览 90人参与
# 五一假期,你打算“躺”还是“卷”? #
24641次浏览 389人参与
# 考研可以缓解求职焦虑吗 #
20343次浏览 241人参与
# 如何缓解入职前的焦虑 #
171545次浏览 1267人参与
# 面试等了一周没回复,还有戏吗 #
115133次浏览 1072人参与
# 找工作前vs找工作后的心路变化 #
7090次浏览 64人参与
# 应届生薪资多少才合理? #
3031次浏览 24人参与
# 写简历别走弯路 #
714026次浏览 7848人参与
# 你喜欢工作还是上学 #
37249次浏览 407人参与
# 如果有时光机,你最想去到哪个年纪? #
43163次浏览 765人参与
# 牛友们的论文几号送审 #
27121次浏览 623人参与
# 扒一扒那些奇葩实习经历 #
41437次浏览 770人参与
# 24届的你们现状如何了? #
64477次浏览 377人参与