[NEO解题报告]《Leetcode》面试题 17.14. 最小K个数 -- ⭐-CSDN博客

本文链接：https://blog.csdn.net/quner6/article/details/120092311

文章目录

1. 题目信息
- 1.1 题目描述
- 1.2 测试用例
2. 题目分析
- 2.1 sort后取前k个
- 2.2 大根堆存最小的K个
3. 代码详情
4. 系列文章

1. 题目信息

1.1 题目描述

题目链接：面试题 17.14. 最小K个数

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

1.2 测试用例

示例：

输入： 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))

2. 题目分析

2.1 sort后取前k个

很直观的思路，排序后，取最小的 K 个

2.2 大根堆存最小的K个

C++中的大根堆，就是 priority_queue , 默认 top 存放最大元素；

先存前面 K 个元素到堆中；
遇到新的数字和堆里面的最大值即 top 进行比较，如果当前值更小，则入堆，替换掉原来的最大值。
最后剩下的元素放入数组中返回即可；

3. 代码详情

3.1 C++

3.1.1 sort方法

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

3.1.2 大根堆方法

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        // 执行用时：36 ms, 在所有 C++ 提交中击败了46.77%的用户
        // 内存消耗：18.2 MB, 在所有 C++ 提交中击败了37.65%的用户
        if (k == 0) {
            return vector<int>{};
        }
        priority_queue<int> small;
        for (int i : arr) {
            if (small.size() < k) {
                small.push(i);
            } else {
                if (i < small.top()) {
                    small.pop();
                    small.push(i);
                }
            }
        }
        vector<int>ans(k);
        int cnt = 0;
        while(big.size()) {
            ans[cnt++] = big.top();
            big.pop();
        }
        return ans;
    }
};