LeetCode——面试题 17.14. 最小K个数(Smallest K LCCI)[中等]——分析及代码（C++）

一、题目

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

示例：

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

提示：

• 0 <= len(arr) <= 100000
• 0 <= k <= min(100000, len(arr))

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-k-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 排序

（1）思路

直接对数组中的所有元素进行排序，然后输出最小的 k 个数。

时间复杂度为 O(nlogn)。

（2）代码

class Solution {
   
public:
    vector<int> smallestK(vector<int>& arr, int k) {
   
        sort(arr.begin(), arr.end());//排序
        vector<int> ans(k);
        for (int i = 0; i < k; i++)//找出最小的k个数
            ans[i] = arr[i];
        return ans;
    }
};

（3）结果

执行用时 ：28 ms，在所有 C++ 提交中击败了 76.24% 的用户；
内存消耗 ：17.5 MB，在所有 C++ 提交中击败了 91.23% 的用户。

2. 堆（优先队列）

（1）思路

设计一个大顶堆（优先队列），用于存放最小的 k 个数。在遍历数组中所有元素时，若堆中元素个数小于 k，可直接将数字放入堆中；此外若数字小于堆顶元素，则在弹出堆顶的同时将该数字入堆。最终堆中的元素就是 k 个最小的数字。

时间复杂度为 O(n)。

（2）代码

class Solution {
   
public:
    vector<