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


你可能感兴趣的:(算法,C++)