剑指offer 面试题40 最小的K个数

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

tips: 使用C++ STL 中的multiset,维护一个O(k)大小的红黑树,时间复杂度O(nlog(n))
学习了set, multiset的基本操作以及内部原理(红黑树)

typedef multiset<int,greater<int> > IntSet;
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(k<=0||input.size()<k) {
            return res;
        }
        IntSet myset;
        for (int i = 0; i < input.size(); i++)
        {
            if(myset.size()<k) {
                myset.insert(input[i]);
            } else {
                auto iter=myset.begin();
                if(input[i]<(*iter)) {
                    myset.erase((*iter));
                    myset.insert(input[i]);
                }
            }
        }
        
        for (auto iter = myset.end(); iter!= myset.begin(); iter--)
        {
            res.push_back((*iter));
        }
        return res;
    }
};

你可能感兴趣的:(剑指offer)