码迷,mamicode.com
首页 > 其他好文 > 详细

【剑指Offer】最小的K个数

时间:2019-08-30 21:03:24      阅读:76      评论:0      收藏:0      [点我收藏+]

标签:==   nlog   题目   cpp   个数   set   ima   solution   tor   

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

 

A1:先把数组排序,然后输出前面的K个数   ====> O(nlogn)

A2:当可以修改数组时可以,把第k个数为基准,把所有比第k个数字小的所有数字位于数组的作变,比第k个数字大的所有数字位于数组的右边

   ====> O(n) 但这不是排序的

A3:用最大堆 || 红黑树 作为容器,每次可以在O(1)时间得到已有的k个数字中的最大值,但需要O(logk)时间完成删除及插入操作

  ====> O(nlogk)

  set和multiset都是基于红黑树实现的

 

//排序   O(nlogn)

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

技术图片

 

【剑指Offer】最小的K个数

标签:==   nlog   题目   cpp   个数   set   ima   solution   tor   

原文地址:https://www.cnblogs.com/xiexinbei0318/p/11436975.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!