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

     }

}