【练习】查找最小的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]<<" "; } }