最小的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;
}