[剑指offer] 最小的K个数，C++实现

  1 // 使用红黑树实现
  2 class Solution {
  3 public:
  4     vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
  5         // 特殊输入
  6         if(input.size()<=0||k>input.size()) return vector<int>();
  7 
  8         // 建立用于存储结果的多重集合及多重集合的最大元素的迭代器
  9         multiset<int, greater<int> > res;
 10         multiset<int, greater<int> >::iterator res_best;
 11 
 12         // 遍历输入
 13         vector<int>::iterator input_ite = input.begin(); // 向量迭代器
 14         for(;input_ite!=input.end();input_ite++)
 15         {
 16             // 将前k个元素插入集合
 17             if(res.size()<k)
 18                 res.insert(*input_ite);
 19             else
 20             {
 21                 // 多重集合中最大元素的迭代器
 22                 res_best = res.begin();
 23                 // 比较+删除+插入
 24                 if(*input_ite<*(res.begin()))
 25                 {
 26                     res.erase(res_best);
 27                     res.insert(*input_ite);
 28                 }
 29             }
 30         }
 31         return vector<int>(res.begin(),res.end());
 32     }
 33 };