力扣-面试题 17.14. 最小K个数（C++）- 排序法+大根堆法

最新推荐文章于 2024-09-30 07:31:33 发布

JIngles123

最新推荐文章于 2024-09-30 07:31:33 发布

阅读量189

点赞数

分类专栏： # 中等题文章标签： c++ 面试 leetcode

本文链接：https://blog.csdn.net/qq_40467670/article/details/120097338

版权

中等题专栏收录该内容

177 篇文章 0 订阅

订阅专栏

这篇博客介绍了两种方法来解决LeetCode中的最小k个数问题。解一采用排序法，直接对数组进行排序然后返回前k个元素。解二利用大根堆，维护一个大小为k的堆，依次将数组元素插入堆中，如果新元素小于堆顶元素则替换，最后得到的堆即为最小的k个数。这两种方法分别展示了排序和堆在解决此类问题上的应用。

摘要由CSDN通过智能技术生成

题目链接：https://leetcode-cn.com/problems/smallest-k-lcci/
题目如下：
在这里插入图片描述
解一：排序法

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

解二：大根堆

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        //C++中的堆（优先队列）为大根堆
        vector<int> result(k,0);//不能少(k,0)！！！因为返回的时k个元素的数
        if(k==0) return result;

        priority_queue<int> que;
        //目标：只在大根堆中存放k个元素，故此，从k+1个位置开始，但凡小于大根堆堆顶元素的，则需移除在放入当前元素
        for(int i=0;i<k;i++)
        que.push(arr[i]);

        for(int i=k;i<(int)arr.size();++i)
        if(que.top()>arr[i]){
            que.pop();
            que.push(arr[i]);
        }

        for(int i=0;i<k;i++){
            result[i]=que.top();
            que.pop();
        }

        return result;
    }
};