题目:
//输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路:
采用优先级队列 默认大顶堆 其中元素个数超过k时,每次加入一个元素就删除一个头元素
时间复杂度为O(nlog(k+1))
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
{
if (k>input.size())
return vector<int>();
vector<int> tmp;
priority_queue<int> que(input.begin(), input.begin()+k);
for (int i=k;i<input.size();i++)
{
que.push(input[i]);
que.pop();
}
while(!que.empty())
{
tmp.push_back(que.top());
que.pop();
}
return vector<int> (tmp.rbegin(), tmp.rend());
}
};