//可以用基於vector的heap,或者使用基於紅黑樹的set, 兩者都給出
#include <vector>
#include <set>
#include <algorithm>
#include <iostream>
using namespace std;
int ia[9] = {0, 9, 8, 3, 4, 1, 2, 7, 0};
void UseHeap(int k) {
vector<int> ivec(ia, ia+9);
if(k > ivec.size())
return;
vector<int>::iterator iter = ivec.begin();
make_heap(iter, iter+k);
for(iter = iter+k; iter != ivec.end(); ++iter) {
if (*iter < *(ivec.begin())) {
iter_swap(iter, ivec.begin());
pop_heap(ivec.begin(), ivec.begin()+k);
push_heap(ivec.begin(), ivec.begin()+k);
}
}
for(int i=0; i<k; ++i)
cout<< ivec[i]<<" ";
cout<<"finish Heap"<<endl;
}
void UseSet(int k) {
multiset<int> mset;
mset.clear();
multiset<int>::iterator mter;
vector<int> ivec(ia, ia+9);
vector<int>::iterator iter = ivec.begin();
for(; iter!=ivec.end(); ++iter) {
if (k > mset.size()) {
mset.insert(*iter);
}
else {
mter = mset.end();
--mter;
if (*iter < *mter) {
mset.erase(mter);
mset.insert(*iter);
}
}
}
for(mter=mset.begin(); mter!=mset.end(); ++mter)
cout<< *mter <<" ";
cout<<"finish set"<<endl;
}
int main() {
UseHeap(5);
UseSet(5);
return 0;
}
找最大的k個數
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.