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

    }

}