微软面试题<五>

题目：

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