剑指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;
     }
 };