题解 | 最小的K个数

#include <functional>
#include <queue>
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param input int整型vector 
     * @param k int整型 
     * @return int整型vector
     */
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        // write code here
        vector<int> ans;

        if (k == 0)
            return ans;

        priority_queue<int, vector<int>, less<int>> pq;

        for (int i = 0; i < input.size(); ++i) {
            if (pq.size() < k) {
                pq.push(input[i]);
            }
            else {
                if (input[i] >= pq.top())
                    continue;
                else {
                    // cout << pq.top() << endl;
                    pq.pop();
                    pq.push(input[i]);
                }
            }
        }

        while(!pq.empty()) {
            ans.push_back(pq.top());
            pq.pop();
        }

        return ans;
    }
};

使用一个大小为 k 的大根堆记录当前最小的 k 个数

若当前元素小于堆顶元素，弹出堆顶元素，将当前元素入堆，堆中最后剩下的元素即为最小的 k 个数