剑指offer40:最小的K个数

最新推荐文章于 2020-11-02 02:12:09 发布

雪不化花不开

最新推荐文章于 2020-11-02 02:12:09 发布

阅读量225

点赞数

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/u011671986/article/details/80051924

版权

数据结构专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用C++实现的高效算法，该算法能够从输入的整数数组中找到最小的K个数。通过使用multiset数据结构，确保了算法的时间效率。文章详细展示了如何迭代输入数组并维护一个大小为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) {
            multiset<int, greater<int>> leastK;
            vector<int> retvec;
            if(k>input.size())
                return retvec;
  
            for(auto in:input)
            {
                if(leastK.size()<k)
                {
                    leastK.insert(in);
                }else
                {
                    auto iter = leastK.begin();

                    if(in < *iter)
                    {
                        leastK.erase(iter);
                        leastK.insert(in);
                     }
                }
            }
            
            for(auto ret: leastK)
            {
                retvec.push_back(ret);
            }
            return retvec;
        }
    };