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;

     }

 };