查找最小的k个元素

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

分析：采取一般的排序算法，如快速排序，堆排序，归并排序，等，先将这n个整数进行排序，然后再取最小的k个。那么时间复杂度为nlgn。

           我们可以从数组中取k个元素，建一个k个元素的最大堆，继续从数组里面取出元素，如果该元素大于堆中最大元素，那么舍弃，继续；如果小于，则将该元素替换掉堆中最大元素，并调整最大堆结构，维护最大堆性质不变，维护最大堆的性质，时间复杂度为lgk,那么总的时间复杂度为nlgk。这样最终堆中的k个元素即为最小的k个数。

           而STL中，set和multiset很好地实现了堆的性质，但是set中不允许存在相同的元素，而multiset则允许，那么我们采用multiset来实现最大堆。

#include "stdafx.h"
#include <vector>
#include <set>
#include <iostream>

using namespace std;

typedef multiset<int,greater<int>> IntSet;
void FindLeastKNumbers_(int a[], IntSet &leastNumbers, int nLength,int k)
{
	//a.clear();
	//如果原数组元素个数小于k,则不合题意，返回
	if(nLength < k)
		return;
	for(int i=0;i<nLength;i++)
	{
		if(leastNumbers.size()<k)
			leastNumbers.insert(a[i]);	//取前面k个元素建最大堆
		else
		{
			IntSet::iterator iterFirst = leastNumbers.begin();
			if(*iterFirst>(a[i]))
			{
				leastNumbers.erase(iterFirst);
				leastNumbers.insert(a[i]);
			}
		}
	}
}
int _tmain(int argc, _TCHAR* argv[])
{
	int a[20] = {1,2,7,9,3,6,2,9,8,10,11,17,12,18,19,16,17,14,5,20};
	IntSet heapset;
	FindLeastKNumbers_(a,heapset,20,6);
	IntSet::iterator iter = heapset.begin();
	for(;iter!=heapset.end();iter++)
		cout<<*iter<<",";
	return 0;
}