首页 > 编程笔记

C++ accumulate函数用法详解

我们已经介绍过 accumulate() 算法的基本版本,可以用 + 运算符求出元素序列的和。前两个参数是定义序列的输入迭代器,第三个参数是和的初值;第三个参数的类型决定了返回值的类型。第二个版本的第 4 个参数是定义应用到总数和元素之间的二元函数对象。这时,我们在必要时可以定义自己的加法运算。例如:
std::vector<int> values {2, 0, 12, 3, 5, 0, 2, 7, 0, 8};
int min {3};
auto sum = std::accumulate(std::begin(values), std::end(values), 0, [min] (int sum, int v)
{
    if(v < min)
        return sum;
    return sum + v;
});
std::cout << "The sum of the elements greater than " << min-1<<"is " << sum << std::endl;  // 35
这里忽略了值小于 3 的元素。这个条件可以尽可能复杂,因此,我们能够求出指定范围内的元素之和。这个运算并不一定要是加法,可以是任何不修改操作数或不使定义范围的迭代器无效的运算。例如,为数值元素定义的乘法运算函数会生成元素的乘积,只要初值为 1。实现浮点元素除法的函数会生成元素乘积的倒数,只要初值为 1。下面展示了如何生成元素的乘积:
std::vector<int> values {2, 3, 5, 7, 11, 13};
auto product = std::accumulate(std::begin(values), std::end(values), 1, std::multiplies<int>()); // 30030
这里用来自于 functional 头文件的函数作为第 4 个参数。如果有值为 0 的元素,可以像上一个代码段中的 lambda 表达式那样忽略它们。

string 类支持加法,因此可以将 accumulate() 应用到 string 对象的序列上:
std:: vector<string> numbers {"one", "two", "three", "four", "five","six", "seven","eight", "nine","ten"};
auto s = std::accumulate(std::begin(numbers), std::end(numbers), string{},[](strings str, string& element)
{
    if (element[0] == 't')
        return str +' '+ element;
    return str;
}); // Result: " two three ten"
这段代码连接了以开头的 string 对象,并用空格将它们隔开。acumulate() 算法得到的结果可能和它所应用的序列中的元素类型不同:
std::vector<int> numbers {1, 2, 3, 10, 11, 12};
auto s = std::accumulate(std::begin(numbers), std::end(numbers),string {"The numbers are"},[](strings str, int n){ return str + " : " + std::to_string(n);});
std::cout << s << std::endl;//Output: The numbers are: 1: 2: 3: 10: 11: 12
lambda 表达式使用的 to_string() 函数会返回一个数值参数的 string 形式,所以应用 accumulate() 到这里的整数序列会返回注释中显示的 string。

推荐阅读