【剑指Offer】最小的K个数

题目:输入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;
    }
};

 

posted @ 2019-08-30 20:32  XieXinBei0318  阅读(160)  评论(0编辑  收藏  举报