微软面试题<五>

题目：

查找最小的k个元素
题目：输入n个整数，输出其中最小的k个。
例如输入1，2，3，4，5，6，7和8这8个数字，则最小的4个数字为1，2，3和4。

这里采用的是STL的最大堆来解决，先取k个元素建堆，然后取剩下的元素与堆中的最大值进行比较，若小于堆中的最大值，则该值进堆，进行若干次循环，即可得到结果.

代码实现：

#include <iostream>
#include <algorithm>
#include <vector>
#include <functional>
using namespace std;
bool Find_K_Min(vector<int> &data,int k);
int main()
{
	int n,x,k;
	cin>>n>>k;
	vector<int> a;
	for(int i=0;i<n;++i)
	{
		cin>>x;
		a.push_back(x);
	}
	Find_K_Min(a,k);
	for(vector<int> ::iterator it1=a.begin();it1!=a.begin()+k;++it1)
	cout<<*it1<<" ";
	system("pause");
	return 0;
}

bool Find_K_Min(vector<int> &data,int k)
{
	if(k==0||data.size()<k)
	return false;
	make_heap(data.begin(),data.begin()+k);
	for(vector<int> ::iterator it=data.begin()+k;it!=data.end();++it)
	{
		if((*it)<(*data.begin()))
		{
			*data.begin()=*it;
			make_heap(data.begin(),data.begin()+k);
		}
	}
	sort_heap(data.begin(),data.begin()+k);
	return true;
}