关注
#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;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
正在热议
# 晒一晒我的offer #
9714541次浏览 104875人参与
# 你觉得第一学历对求职有影响吗? #
14610次浏览 120人参与
# 北方华创开奖 #
21534次浏览 244人参与
# 地方国企笔面经互助 #
2163次浏览 6人参与
# 应届生被毁约被毁意向了怎么办 #
24948次浏览 233人参与
# 学历or实习经历,哪个更重要 #
39471次浏览 285人参与
# 查收我的offer竞争力报告 #
14259次浏览 201人参与
# 机械应届生薪资要多少才合适? #
11988次浏览 58人参与
# 我的工作日记 #
20611次浏览 269人参与
# 寒假躺平还是提前实习 #
57327次浏览 419人参与
# 总结:哪家公司面试体验感最差 #
24019次浏览 117人参与
# 公司情报交流地 #
31252次浏览 220人参与
# 实习,投递多份简历没人回复怎么办 #
2384442次浏览 34217人参与
# 00后45度躺现状 #
36316次浏览 305人参与
# 秋招OC许愿 #
224741次浏览 1857人参与
# 机械人,签完三方你在忙什么? #
23351次浏览 119人参与
# 不给转正的实习,你还去吗 #
1513335次浏览 16942人参与
# 选完offer后,你后悔学本专业吗 #
7464次浏览 47人参与
# 没有实习经历,还有机会进大厂吗 #
803350次浏览 13777人参与
# 得物求职进展汇总 #
62907次浏览 663人参与
# 秋招拿一个offer可以躺平吗 #
103590次浏览 805人参与