面试题40. 最小的k个数

题目：

 

 

解答：

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) 
    {
        vector<int> res;
        priority_queue<int> q;
        for (int a : arr) 
        {
            q.push(a);
            if (q.size() > k)
            {
                q.pop();
            }
        }
        while (!q.empty()) 
        {
            res.push_back(q.top());
            q.pop();
        }
        return res;

    }
};