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