剑指offer—寻找数组最小的k个数

题目：

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

分析：

传统的做法可以是o(nlog(n))，就是先排序然后从前面中取出k个值。高效的做法可以实现o（nlogk），
借助一个容器，容量k，将数组的读入k个数存放在容器中，然后继续访问数组后面的数，将其后每个数和容器中最大值比较，
比最大值小则剔除容器中最大值，同时将该数放入容器！这样就可以保证遍历整个数组之后，容器中的数就是要找的k个较小数。

这种算法策略非常适合大数据，并且可以不必要将整个数组加载入内存！容器要支持具有排序功能，选择C++的multiset！

#include<vector>

#include<iostream>
#include<set>
#include<algorithm>
#include<functional>
using namespace std;

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
	int len = input.size();
	if (len <= 0 || k>len)
		return vector<int>();
	multiset<int, greater<int>> leastNums;//红黑树二叉树，降序(最大堆)
	vector<int>::iterator vec_it = input.begin();
	for (; vec_it != input.end(); vec_it++)
	{
		if (leastNums.size()<k)//填入容器k个值
			leastNums.insert(*vec_it);
		else
		{
			multiset<int, greater<int>>::iterator greast_it = leastNums.begin();//每一种容器就需要特定的迭代器
			if (*vec_it<*greast_it)
			{
				leastNums.erase(greast_it);//弹出容器最大值
				leastNums.insert(*vec_it);//将当前数组值插入
			}
		}
	}
	return vector<int>(leastNums.begin(), leastNums.end());
}
int main()
{
	vector<int> input = {2,4,5,7,89,1,0};
	vector<int> least_k=GetLeastNumbers_Solution(input,5);
	vector<int>::iterator it = least_k.begin();
	for (; it != least_k.end(); it++)
	{
		cout << *it<<endl;
	}
	while (true)
	{

	}
	return 0;
}