牛客网[最小的k个数]

最新推荐文章于 2023-03-29 17:48:47 发布

原创最新推荐文章于 2023-03-29 17:48:47 发布 · 212 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#最小k个数

C++ 专栏收录该内容

62 篇文章

订阅专栏

博客围绕输入n个整数找出最小的K个数展开，给出示例，如输入4,5,1等8个数字，最小的4个是1,2,3,4。还介绍两种解法，一是利用最大堆，时间复杂度O（nlogk）；二是利用全排序，时间复杂度O（nlogn），同时提醒要考虑k<=0的情况。

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

题目描述

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

解法1；利用最大堆，O（nlogk）

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        if(input.empty()||k>input.size()||k<=0)
            return {};
        vector<int>heap_input(input.begin(),input.begin()+k);
        make_heap(heap_input.begin(),heap_input.end());
        for(int i=k;i<input.size();i++)
        {
            if(input[i]<*heap_input.begin())
            {  pop_heap(heap_input.begin(),heap_input.end());
                heap_input.pop_back();
             heap_input.push_back(input[i]);
             push_heap(heap_input.begin(),heap_input.end());
            }
        }    
        return heap_input;
       
    }

注意题目条件要考虑周期，对于k<=0这种情况开始没有考虑，导致在牛客网上编译通不过。

解法2：利用全排序时间复杂度O（nlogn）

 vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        if(input.empty()||k>input.size())
            return {};
        vector<int>res;
        sort(input.begin(),input.end());
        for(int i=0;i<k;i++)
        {
            res.push_back(input[i]);
        }
        return res;
    }