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

从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);
            }
        }
    }   
}

你可能感兴趣的:(Leetcode)