c++ - 获取 C++ 多重集中的前 N ​​个元素

Question

如何从多重集结构中获取前 N 个元素，而不是不断地获取第一个 (.begin()) 元素然后将其删除？

我只想在不影响多重集的情况下对前 N 个元素求和。

Answer 1

I just want to sum the first N elements without affecting the multiset.

#include <numeric>
#include <iterator>

// ...

int sum = std::accumulate(my_set.begin(), std::next(my_set.begin(), N));

std::next 是 C++11 库的补充。这是针对旧编译器的解决方案:

std::multiset<int>::iterator it = my_set.begin();
std::advance(it, N);
int sum = std::accumulate(my_set.begin(), it);

两种解决方案都对多重集进行了两次迭代。如果你想防止这种情况发生，请使用手动循环:

int sum = 0;
std::multiset<int>::iterator it = my_set.begin();
for (int i = 0; i < N; ++i)
{
    sum += *it++;
}