找最大的k个数

//可以用基于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;
}