29、剑指offer--最小的K个数

题目描述

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

 

解题思路：使用multiset存储k个最小值

1）先存入k个值

2）用multiset中的最大值和当前访问数组元素比较，若小于则把该值从multiset中移除，数组元素插入

3）遍历multiset将k个值存入vector中
注意事项：边界条件的判断，数组为空，k小于1，以及k大于数组元素数目

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> result;
        if(input.empty() || k<1 || input.size()<k)
            return result;
        multiset<int> insert_set;
 
        set<int>::iterator it; //定义前向迭代器
        multiset<int>::reverse_iterator rit; //定义反向迭代器
        for(int i=0;i<input.size();i++)
        {
 
            if(i<k)
            {
                insert_set.insert(input[i]);
            }
 
            else
            {
                rit = insert_set.rbegin();
                if(input[i] < *rit)
                {
 
                    insert_set.erase(*rit);
                    insert_set.insert(input[i]);
                }
            }
        }
 
 
        for(it = insert_set.begin(); it != insert_set.end(); it++)
        {
            result.push_back(*it);
        }
        return result;
    }
};