剑指Offer——（29）最小的k个数

题目描述：

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

实现如下：

//测试用例中给出的返回值是有序的。。啧啧。。
//找到序列中最小的k个数，利用set，对数字进行insert或erase
//红黑树中查找、插入、删除操作都为 O（logk）
//最终将set中存储的值返回值vector中
//方法不改变原vector
//注意：
//1.判断传入的vector是否为空
//2.判断传入的k是否合法
class Solution 
{
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) 
    {
        vector<int> output;
        if (input.size() <= 0 || k <= 0 || k > input.size()) return output;//防御性动作
        set<int> se;
        set<int>::iterator it;
        for (int i = 0; i < input.size(); ++i)//遍历原vector中的所有数
        {
            if (i < k)//如果set中的个数小于k，则直接插入
                se.insert(input[i]);
            else//否则将set中的最大值和vector中第i个数比较
            {
                it = --se.end();//取set中的最大值
                if (*it > input[i])//若set中的最大值大于vector中第i个数
                {
                    se.erase(it);//删除set中的最大值
                    se.insert(input[i]);//插入vector中第i个数
                }
            }
        }

        it = se.begin();
        for (; it != se.end(); ++it)//将set中的数以vector形式return
        {
            output.push_back(*it);
        }
        return output;
    }
};