剑指 Offer 40. 最小的k个数

无名小卒一枚

于 2021-02-02 20:20:59 发布

阅读量76

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：最小堆 K最小

本文链接：https://blog.youkuaiyun.com/sdau_clt/article/details/113573344

LeetCode 专栏收录该内容

210 篇文章

订阅专栏

本文介绍了一种使用最小堆的数据结构来找出整数数组中最小的k个数的方法。通过C++实现，首先创建最小堆，然后依次取出最小元素直至找到k个数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

C++


class Solution {

public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
            make_heap(arr.begin(),arr.end(),greater<int>());  //创建最小堆
            vector<int> out; //返回向量
            //取最小的k个数
            for (int i =1;i<=k;i++){

                pop_heap(arr.begin(),arr.end(),greater<int>());  
                out.push_back(arr.back());
                arr.pop_back();  
            }

            return out;

    }
};