剑指offer：最小的K个数

时间限制：1秒 空间限制：32768K
本题知识点： 数组
题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

/*思路一：
*O(n*logn)
*排序->取前k位
*思路二：
*O(n*log k)
*遍历长度为n的数组，对最小的k个数进行维护
*我想的是通过优先级队列（堆实现）对数组进行存入，然后取出k个最小元素，时间复杂度即为O（n*logk）
*/

代码：
C++:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(input.size() == 0 || k > input.size())
            return res;
        priority_queue<int, vector<int>, greater<int> >pq; 
        for(auto e : input){
            pq.push(e);
        }
        for(int i = 0; i < k; i++){
            res.push_back(pq.top());
            pq.pop();
        }
        return res;
    }
};