找最大的k个数

最新推荐文章于 2019-10-09 22:37:49 发布

游鮕子

最新推荐文章于 2019-10-09 22:37:49 发布

阅读量334

点赞数

分类专栏：语言算法 stl 文章标签： heap stl set

本文链接：https://blog.csdn.net/lh648365878/article/details/47777909

版权

算法同时被 3 个专栏收录

14 篇文章 0 订阅

订阅专栏

语言

5 篇文章 0 订阅

订阅专栏

stl

5 篇文章 0 订阅

订阅专栏

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