class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
priority_queue<int> max_heap;
vector<int> results;
if(input.size() < k || k<=0)
return results;
for(auto& val: input)
{
// 堆还不够最小的容量,直接加进去
if(max_heap.size() < k) {
max_heap.push(val);
}
//让当前元素与队顶元素相比
else if (max_heap.top() > val) {
max_heap.push(val);
// 当前元素入队
max_heap.pop();
}
}
while(max_heap.size())
{
results.push_back(max_heap.top());
max_heap.pop();
}
reverse(results.begin(), results.end());
return results;
}
};
剑指数据结构-最小的K个数
最新推荐文章于 2024-04-23 21:26:48 发布