C++常类型(const 2)

寻找最小的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 ;    
}



C++常类型(const 2)

上一篇:C语言中extern的用法--转


下一篇:优秀Python学习资源收集汇总(强烈推荐)