从n个数中,找出最小的k个数

来源:互联网 发布:五级三阶制算法 编辑:程序博客网 时间:2024/05/21 03:28

从n个数中,找出最小的k个数
set,multiset是基于红黑树排列的,.begin()是最小值。
在头文件中加入functional.

void FindNums(const vector<int> &data, multiset<int, greater<int>> &m, int k){    if (k < 1)        throw "error";    if (k>data.size())            return;    auto iter = data.cbegin();    for (; iter != data.cend(); ++iter)    {        if (k >m.size())            m.insert(*iter);//插入        else        {            auto greaterNum = m.begin();//基于红黑树的最大到小排            if (*iter < *greaterNum)            {                m.erase(greaterNum);                m.insert(*iter);            }        }    }   }
0 0