寻找最小的k个数
题目描述:5.查找最小的k个元素
题目:输入n个整数,输出其中最小的k个。
例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。
堆实现可见:简洁的heap代码
#include <iostream>
#include <windows.h>
#include <algorithm>
#include <numeric>
#include <string>
#include <array>
#include <vector>
#include <functional>
#include <hash_set>
#include <ctime>
using namespace std;
#define N 20//最大的N个数
//第一种方法,维护N+1个元素的小顶堆
//O(NlogK)
void f1(){
//freopen("C:\\in.txt","r",stdin);
vector<int> t(10000);
srand((unsigned)time(NULL));
generate(t.begin(),t.end(),[](){return rand()%10000;});
make_heap(t.begin(),t.begin()+N+1,greater<int>());
for(auto it=t.begin()+N+1;it!=t.end();){
push_heap(t.begin(),t.begin()+N+1,greater<int>());
pop_heap(t.begin(),t.begin()+N+1,greater<int>());
it=t.erase(t.begin()+N);
}
for_each(t.begin(),t.end(),[](int i){cout<<i<<endl;});
}
//第二种方法:整个建堆,pop K次
//O(N+KlogN)
void f2(){
vector<int> t(1000000);
srand((unsigned)time(NULL));
generate(t.begin(),t.end(),[](){return rand()%10000;});
make_heap(t.begin(),t.end());//对所有元素建大顶堆
for(int i=0;i<N;i++){
cout<<t.front()<<endl;
pop_heap(t.begin(),t.end());
t.pop_back();
}
}
void Test(){
f2();
}
int main()
{
LARGE_INTEGER BegainTime ;
LARGE_INTEGER EndTime ;
LARGE_INTEGER Frequency ;
QueryPerformanceFrequency(&Frequency);
QueryPerformanceCounter(&BegainTime) ;
//要测试的代码放在这里
Test();
QueryPerformanceCounter(&EndTime);
//输出运行时间(单位:s)
cout << "运行时间(单位:s):" <<(double)( EndTime.QuadPart - BegainTime.QuadPart )/ Frequency.QuadPart <<endl;
//system("pause") ;
return 0 ;
}