【练习】查找最小的k 个元素

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

思路：利用make_heap选出vector前K个数的最大值，对于后面的元素，与最大值作比较，如果小于最大值，则替换最大值，然后再次make_heap，选出当前k个数中的最大值.
此种解法利用STL函数跳过了建堆过程。
//make_heap是标准算法库里的模板函数,用于将存储在vector/deque 中的元素进行堆操作,将[start, end)范围进行堆排序，默认使用less<int>, 即最大元素放在第一个,也即大根堆。
*/
/************************************************************************/ 

#include <iostream>
#include <vector>
#include <algorithm>
#include <functional>

using namespace  std;
void find(vector<int> &a,int k);
int main()
{

	vector<int> a;
	for (int i=8;i>0;i--)
	{
		a.push_back(i);
	}
	
	find(a,4);
	return 0;
}

void find(vector<int> &a,int k)
{
	if (a.size()<k)
	{
		return;
	}
	make_heap(a.begin(),a.begin()+k);
	for (vector<int>::size_type i=k;i !=a.size();i++)
	{
		if (a[i]<a[0])
		{
			a[0]=a[i];
			make_heap(a.begin(),a.begin()+k);
		}

	}

	sort(a.begin(),a.begin()+k);
	for (vector<int>::size_type i=0;i !=k;i++)
	{
		cout<<a[i]<<" ";
	}
}

　　

转载于:https://www.cnblogs.com/caidaxia/archive/2011/10/18/2216144.html