std::partial_sum用于计算前缀和或广义累积运算,支持自定义二元操作;可原地计算但需输入为RandomAccessIterator且输出起始≥输入起始;C++17起可用std::inclusive_scan替代,支持并行。
std::partial_sum 计算前缀和的基本用法
std::partial_sum 是 C++ 标准库中 头文件提供的算法,用于就地或复制生成前缀和(或广义累积运算结果)。它不只限于加法——只要提供二元操作符,就能做前缀积、最大值传播、自定义折叠等。
最常见用法是计算累加前缀和:
std::vector v = {1, 2, 3, 4};
std::vector result(v.size());
std::partial_sum(v.begin(), v.end(), result.begin()); // result = {1, 3, 6, 10}
注意:输出迭代器可以和输入重叠(如原地计算),但必须保证不会发生未定义行为——result.begin() 不能在 v.begin() 前面,否则写入会覆盖尚未读取的输入值。
为什么不能直接用 std::partial_sum(v.begin(), v.end(), v.begin())?
可以,但有前提:必须确保容器支持安全的原地计算。对 std::vector、std::deque 等连续/随机访问容器,std::partial_sum(v.begin(), v.end(), v.begin()) 是合法且高效的;但对 std::list 或输入迭代器范围(如 std::istream_iterator),输出不能指向输入源的同一范围,否则行为未定义。
立即学习“C++免费学习笔记(深入)”;
常见错误现象:std::partial_sum 在原地操作时看似结果正确,但在某些优化级别下出现乱值——往往是因为误用了不满足“可安全重叠”条件的迭代器类型。
- 安全原地使用:输入为
RandomAccessIterator,且输出起始位置 ≥ 输入起始位置 - 不安全场景:用
std::partial_sum(it1, it2, it0),其中it0 (例如想往前偏移写入) - 替代方案:若需错位写入(如前缀和右移一位),先用临时容器,再移动或拷贝
用自定义二元操作实现前缀最小值、乘积等
std::partial_sum 第四个参数接受任意二元函数对象,因此不只是加法。比如求前缀最小值: