LeetCode刷题笔记——40. 最小的k个数（最大堆、priority_queue、vector）

题目

1. 最大堆

2. priority_queue

3. vector

4. 答题

参考来源：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/solution/zui-xiao-de-kge-shu-by-leetcode-solution/

1. 对原数组从小到大排序后取出前 k个数即可。

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        sort(arr.begin(), arr.end());//sort from small to big
        for(int i = 0; i < k; i++)
            vec[i] = arr[i];
        return vec;
    }
};

2. 我们用一个大根堆实时维护数组的 k个小值。首先将前 k 个数插入大根堆中，随后从第 k+1个数开始遍历，如果当前遍历到的数比大根堆的堆顶的数要小，就把堆顶的数弹出，再插入当前遍历到的数。最后将大根堆里的数存入数组返回即可。

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        if(k==0)   //排除k为0的情况
            return vec;
        priority_queue<int> Q; //实例化最大堆
        for(int i = 0; i < k; i++)
            Q.push(arr[i]); //先push进arr前k个数
        for(int i = k; i < (int)arr.size(); i++)//处理k后面的数字
        {
            if(Q.top() > arr[i])
            {
                Q.pop();
                Q.push(arr[i]);
            }
        }
        for(int i = 0; i < k; i++){
            vec[i] = Q.top();
            Q.pop();//输出top并pop
        }
        return vec；
    }
};