题目:库存管理 III

使用内置的函数 nth_element()

难绷

inventoryManagement()

1vector<int> inventoryManagement(vector<int>& stock, int cnt)
2{
3    if(stock.empty() || cnt == 0) return {};
4
5    std::nth_element(stock.begin(), stock.begin() + cnt, stock.end());
6
7    std::vector<int> res(stock.begin(), stock.begin() + cnt);
8    return res;
9}

原文:easy