最小的K个数

标签：c++   刷题   

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

思路：

    我们可以先创建一个大小为k的数据容器来存储最小的k个数字，接下来我们每次从输入的n个整数中读入一个数。如果容器中已有的数字少于k个，则直接把这次读入的整数放入容器之中;如果容器中己有k个数字了，也就是容器已满，此时我们不能再插入新的数字而只能替换已有的数字。找出这已有的k个数中的最大值，然后拿这次待插入的整数和最大值进行比较。如果待插入的值比当前己有的最大值小，则用这个数替换当前已有的最大值;如果待插入的值比当前已有的最大值还要大，那么这个数不可能是最小的k个整数之一，于是我们可以抛弃这个整数。方法适合海量数据的处理。

#include<iostream>
#include <set>

typedef multiset<int,greater<int>> intSet;
typedef multiset<int,greater<int>>::iterator SetIterator;

void GetLeastNumbers(const vector<int> &data, intSet &LeastNumbers, int k)
{
	LeastNumbers.clear();
	//判断输入参数
	if (k<1||data.size()<k)
		return;
	//定义一个迭代器
	vector<int>::const_iterator iter=data.begin();
	//循环
	for ( ;iter!=data.end(); iter++)
	{
		if ((LeastNumbers.size())<k)//容器中已有数字少于k，读入
		{
			LeastNumbers.insert(*iter);
		}
		else
		{
			//指向容器中k个数字中最大数
			SetIterator iterGreatest=LeastNumbers.begin();
			if (*iter<*(LeastNumbers.begin()))
			{
				LeastNumbers.erase(iterGreatest);
				LeastNumbers.insert(*iter);
			}
		}
	}
}

最小的K个数

标签：c++   刷题   

原文地址：http://blog.csdn.net/lsh_2013/article/details/46050931