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