面试题 17.14. 最小K个数

😀最近新创建了个开源仓库，总结LeetCode的每日一题，目前已有C++、JavaScript语言版本，欢迎大家提供其他语言版本！ 🩲仓库地址：每日一题系列

题目描述：

设计一个算法，找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例：

输入： arr = [1,3,5,7,2,4,6,8], k = 4
输出： [1,2,3,4]

解答：

C++：

方法一

优先队列 可以看295. 数据流的中位数

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        /*sort(arr.begin(),arr.end());
        return vector<int>(arr.begin(),arr.begin()+k);*/
        //优先队列
        if(k==0) return {};
        priority_queue<int> res;
        for(auto &x:arr){
            if(res.size()<k){
                res.push(x);
            }else{
                if(x<res.top()){
                    res.pop();
                    res.push(x);
                }
            }
        }
        vector<int> ans;
        while(!res.empty()){
            ans.push_back(res.top());
            res.pop();
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
};

方法二

排序

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        sort(arr.begin(),arr.end());
        return vector<int>(arr.begin(),arr.begin()+k);
    }
};