剑指offer——42最小的K个数

题目描述

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

 

题解：
　　原以为书中会有好方法，想不到还是排序和STL这两种方法。

　　

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        if (input.size() < k)return {};
        set<int>res;
        for (auto a : input)
        {
            if (res.size() < k)
                res.insert(a);
            else if(a<*(--res.end()))
            {
                res.erase(--res.end());
                res.insert(a);
            }
        }
        vector<int>v;
        v.assign(res.begin(), res.end());
        return v;
    }
};