最小的k个数--适合处理海量数据

#include <iostream>
#include <set>//multiset所属头文件
using namespace std;

#define K 4

void find_number(int a[], int len, multiset<int,greater<int> > &set_number, int k)
{
	set_number.clear();
	if(a==NULL || len <=0 || k<=0 || k>len)
		return;
	int i;
	for(i=0;i<len;i++)
	{
		if(set_number.size() < k)//堆大小小于k，插入数组值
			set_number.insert(a[i]);
		else
		{
			multiset<int,greater<int> >::iterator iter = set_number.begin();
			if(a[i]<*iter)//如果堆的堆顶也就是堆的最大值大于当前数组值，则堆顶值舍弃，插入当前数组值，并重新堆内排序
			{
				set_number.erase(iter);
				set_number.insert(a[i]);
			}
		}
	}
}

int main()
{
	int a[] = {4,5,1,6,2,7,3,8};
	int len = sizeof(a)/sizeof(int);
	int i;
	cout<<"数组值为:"<<endl;
	for(i=0;i<len;i++)
		cout<<a[i]<<" ";
	cout<<endl;
	multiset<int,greater<int> > set_number; //greater<int>代表内置排序从大到小，另外注意> >中间有空格，否则会被编译器认为是>>
	find_number(a,len,set_number,K);
	cout<<"最小K个数:"<<endl;
	multiset<int,greater<int> >::iterator iter = set_number.begin();
	for(;iter!=set_number.end();iter++)
		cout<<*iter<<" ";
	cout<<endl;
	return 0;
}