std::partial_sum用于计算前缀和或广义累积运算,支持自定义二元操作;可原地计算但需输入为RandomAccessIterator且输出起始≥输入起始;C++17起可用std::inclusive_scan替代,支持并行。

std::partial_sum 计算前缀和的基本用法
std::partial_sum 是 C++ 标准库中 头文件提供的算法,用于就地或复制生成前缀和(或广义累积运算结果)。它不只限于加法——只要提供二元操作符,就能做前缀积、最大值传播、自定义折叠等。
最常见用法是计算累加前缀和:
std::vectorv = {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 第四个参数接受任意二元函数对象,因此不只是加法。比如求前缀最小值:
std::vectora = {5, 3, 8, 1, 9}; std::partial_sum(a.begin(), a.end(), a.begin(), [](int a, int b) { return std::min(a, b); }); // a becomes {5, 3, 3, 1, 1}
关键点在于:该二元操作的**第一个参数是「当前累积结果」,第二个是「下一个输入元素」**,顺序不能反。这和 std::accumulate 的参数顺序一致。
性能影响:若操作本身开销大(如字符串拼接、复杂对象比较),应考虑是否真需每一步都保存中间结果;前缀积还可能迅速溢出,std::partial_sum 不做数值检查。
- 前缀积:用
std::multiplies{}或 lambda - 逻辑与传播:
[](bool a, bool b) { return a && b; } - 注意:操作必须满足结合律吗?不强制,但不符合直觉的结果往往源于非结合操作(如减法)
和 std::inclusive_scan 的区别与兼容性考量
C++17 引入了 std::inclusive_scan(在 中),功能与 std::partial_sum 高度重叠,但更现代、支持执行策略(如并行)、明确区分包含/排除语义。两者主要差异:
-
std::partial_sum是 C++98 就有的,所有编译器都支持;std::inclusive_scan要求 C++17 及以上 -
std::inclusive_scan允许传入std::execution::par实现并行前缀和(需配合支持的 STL 实现,如 libc++ 或 MSVC 最新版) - 二者默认行为一致,但
std::inclusive_scan更强调“扫描”语义,std::partial_sum更偏向“累积”历史命名
如果你的项目需兼容老标准或嵌入式环境(无完整 C++17 支持),坚持用 std::partial_sum 没问题;若已用 C++17 且处理大量数据,std::inclusive_scan 是更可扩展的选择。别混用:同一个逻辑在不同平台用不同函数,容易漏掉并行策略或初始化差异。
最容易被忽略的是迭代器有效性边界——无论用哪个函数,输入范围不能为空(first == last 时行为合法,但输出也为空),且输出空间必须足够容纳 std::distance(first, last) 个元素,否则是未定义行为。










