std::partial_sum 是 C++ 标准库中的一个算法,用于计算序列的部分和或其他二元操作的累积结果。这个函数对输入范围内的每个元素应用二元操作,并将结果存储在输出范围中。对于输入序列[a, b, c, …],输出序列将是 [a, a+b, a+b+c, …]`(如果使用默认的加法操作)。
std::partial_sum 的基本实现原理可以描述为以下步骤:
这里是一个简单的 std::partial_sum 实现示例:
#include<iostream>#include<vector>#include<numeric>template<typename InputIt, typename OutputIt, typename BinaryOperation>OutputIt partial_sum(InputIt first, InputIt last, OutputIt d_first, BinaryOperation op) { if (first == last) return d_first; typename std::iterator_traits<InputIt>::value_type sum = *first; *d_first = sum; while (++first != last) { sum = op(sum, *first); *++d_first = sum; } return ++d_first;}int main() { std::vector<int> nums = {1, 2, 3, 4, 5}; std::vector<int> result(nums.size()); partial_sum(nums.begin(), nums.end(), result.begin(), std::plus<int>()); for (int num : result) { std::cout<< num << " "; } return 0;}这个示例中的 partial_sum 函数模板接受输入范围的起始和结束迭代器、输出范围的起始迭代器以及一个二元操作。在这个例子中,我们使用了加法操作(std::plus<int>())来计算输入序列的部分和。


