剑指offer--最小的K个数

最新推荐文章于 2024-04-23 21:26:48 发布

abc_xian

最新推荐文章于 2024-04-23 21:26:48 发布

阅读量75

点赞数

CC 4.0 BY-SA版权

分类专栏： c++

本文链接：https://blog.youkuaiyun.com/abc_xian/article/details/99877156

c++ 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种寻找数组中最小K个数的有效算法，通过使用红黑树（multiset集合）实现，时间复杂度达到O(nlogk)。文章提供了具体的代码实现，包括排序和动态维护最小堆的方法。

摘要生成于 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> vec;
        if(len<k)
            return vec;
        sort(input.begin(),input.end());
        for(int i=0;i<k;i++)
            vec.push_back(input[i]);
        return vec;
    }
};

红黑树：multiset集合利用仿函数改变排序顺序时间复杂度O（nlogk）

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len=input.size();
        if(len<=0||k>len) return vector<int>();
         
        //仿函数中的greater<T>模板，从大到小排序
        multiset<int, greater<int> > leastNums;
        vector<int>::iterator vec_it=input.begin();
        for(;vec_it!=input.end();vec_it++)
        {
            //将前k个元素插入集合
            if(leastNums.size()<k)
                leastNums.insert(*vec_it);
            else
            {
                //第一个元素是最大值
                multiset<int, greater<int> >::iterator greatest_it=leastNums.begin();
                //如果后续元素<第一个元素，删除第一个，加入当前元素
                if(*vec_it<*(leastNums.begin()))
                {
                    leastNums.erase(greatest_it);
                    leastNums.insert(*vec_it);
                }
            }
        }
         
        return vector<int>(leastNums.begin(),leastNums.end());
    }
};