关注
#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;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
02-05 12:02
杭州电子科技大学 前端工程师 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 牛客新年AI问运 #
10472次浏览 148人参与
# 第一次面试 #
1072524次浏览 13734人参与
# 牛客AI体验站 #
17153次浏览 297人参与
# 投递几十家公司,到现在0offer,大家都一样吗 #
329125次浏览 2127人参与
# 你喜欢工作还是上学 #
89882次浏览 887人参与
# 被AI治愈的瞬间 #
91085次浏览 689人参与
# 有必要和同事成为好朋友吗? #
1789次浏览 34人参与
# 虾皮求职进展汇总 #
378294次浏览 2795人参与
# 百度求职进展汇总 #
667641次浏览 6293人参与
# 招聘要求与实际实习内容不符怎么办 #
169501次浏览 926人参与
# 如果不上班,你会去做什么 #
29405次浏览 466人参与
# 非技术岗薪资爆料 #
491340次浏览 3047人参与
# 你找工作的时候用AI吗? #
173771次浏览 892人参与
# 产品薪资爆料 #
158959次浏览 855人参与
# 你觉得什么岗位会被AI替代 #
41673次浏览 282人参与
# 考研失败就一定是坏事吗? #
222547次浏览 1543人参与
# 国企vs私企,你更想去? #
320085次浏览 2530人参与
# 我的求职精神状态 #
431786次浏览 3082人参与
# 秋招想进国企该如何准备 #
127558次浏览 620人参与
# 卷__卷不过你们,只能卷__了 #
42283次浏览 669人参与