剑指Offer编程练习029——最小的K个数

zh921

于 2018-10-29 17:10:21 发布

阅读量177

点赞数

CC 4.0 BY-SA版权

分类专栏： C++ 算法

本文链接：https://blog.youkuaiyun.com/zhao_h/article/details/83508350

C++ 同时被 2 个专栏收录

46 篇文章

订阅专栏

算法

29 篇文章

订阅专栏

本文介绍了一种高效算法，用于从输入的整数数组中找出最小的K个数。通过构建最大堆并进行堆调整，确保堆顶元素始终为当前已处理数中的最大值，从而实现快速筛选。

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

题目描述

输入n个整数，找出其中最小的K个数。

例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k){
        int len = input.size();
        vector<int> result;
        if (len <= 0 || k > len || k <= 0)
            return result;
        for (int i = 0; i < k; ++i)
            result.push_back(input[i]); //result中最开始存入前k个数
        make_heap (result.begin(),result.end()); //建最大堆
        for (int i = k; i < len; ++i){
            if (input[i] < result[0]) { //对后面的数，如果有比堆中最大的数小的就压入堆中
                pop_heap(result.begin(), result.end());
                result.pop_back();
                result.push_back(input[i]);
                push_heap(result.begin(), result.end());
            }
        }
        sort_heap(result.begin(), result.end());
        return result;
    }
};