输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
优先队列实现
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> vec;
if (input.size()<k)
{
return vec;
}
priority_queue<int> p;
for (auto i = 0; i < input.size();++i)
{
p.push(input[i]);
if (p.size()>k)
{
p.pop();
}
}
while (p.size())
{
vec.push_back(p.top());
p.pop();
}
return vec;
}
};