牛客网《剑指Offer》 编程 29.最小的k个数（使用c++提供的set或者multiset）

题目描述

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

解题思路

此次使用不改变原数组的方法。使用c++STL中的set或者multiset来实现。这两个模板底层的实现都是红黑树。即平衡的二叉搜索树。以下是对set和multiset的简单介绍：

STL	说明
set	按照从小到大顺序排列的set，元素不允许重复
set	按照op排列的set，元素不允许重复
multiset	从小到大顺序排列的mulitset，只是元素允许重复
multiset	按照op排列的set，元素允许重复

op如果是less，就是从小到大；op是greater就是从大到小。

遍历待选数组。

当set或者multiset的大小不大于k的时候，可以直接往里面添加元素；

但是当set或者multiset的大小等于k的时候，此时要将当前元素与容器中的最大的元素进行比较。如果大于，则进行下一轮循环；如果小于则将容器中的最大的元素删除掉，将当前元素插入容器。

代码实现

class Solution {
public:
    vector GetLeastNumbers_Solution(vector input, int k) {
        multiset > leastNumber;
        if(k<1||input.size()();
        }
        for(vector::iterator iter=input.begin();iter!=input.end();iter++){
            if(leastNumber.size() >::iterator setBegin=leastNumber.begin();
                if((*iter)<*(leastNumber.begin())){
                    leastNumber.erase(setBegin);
                    leastNumber.insert(*iter);
                }
            }
        }
        vector res(leastNumber.begin(),leastNumber.end());
        return res;
    }
};