如何在 std::transform 中使用向量中的其他值?

cod*_*der 6 c++ algorithm lambda stl

下面的代码只为向量的每个元素加 1:

#include <iostream>
#include <vector>
#include <algorithm>

int main()
{
    // Create vector 1 - print it out
    std::vector<int> v1 {1,2,3,4,5,6,7,8,9};
    for (auto val : v1)
    {
        std::cout << val << " ";
    }
    std::cout << std::endl;

    // Create vector 2 by a transform of vector 1 + 1
    std::vector<int> v2(v1.size());
    std::transform(v1.begin(), v1.end(), v2.begin(),
        [](int val){ return val + 1; });

    // Print out vector 2
    for (auto val : v2)
    {
        std::cout << val << " ";
    }
    std::cout << std::endl;        
}


Run Code Online (Sandbox Code Playgroud)

我不想只为每个值加 1,而是想添加前一个元素的值,以便:

v[0] = v[0] (no previous value)
v[1] = v[1] + v[0]
v[2] = v[2] + v[1]
 etc...
Run Code Online (Sandbox Code Playgroud)

所以我最终得到:

1     = 1
2 + 1 = 3
3 + 2 = 5
4 + 3 = 7
etc...
Run Code Online (Sandbox Code Playgroud)

这是一个愚蠢的例子 - 它不是数学难题或类似的东西,我只是想弄清楚我是否可以在向量中使用除了传递到 lambda 函数的值之外的其他值?

我在 lambda 中缺少两位数据:

v[0] = v[0] (no previous value)
v[1] = v[1] + v[0]
v[2] = v[2] + v[1]
 etc...
Run Code Online (Sandbox Code Playgroud)

我可以通过捕获传递 v1 并用它来进行一些计算,但不完全是我想要的,因为我不知道索引。

1     = 1
2 + 1 = 3
3 + 2 = 5
4 + 3 = 7
etc...
Run Code Online (Sandbox Code Playgroud)

也许转换不是这项工作的工具,我需要诉诸迭代器循环?:

    std::transform(v1.begin(), v1.end(), v2.begin(), [](int val){
            // What is the index of val?
            // What is the size of the vector v1?
            // How would I express this:
            if (INDEX_OF(val) > 0)
                return val + INDEX_OF(val - 1);
            else
                return val;
        });
Run Code Online (Sandbox Code Playgroud)

这是可行的,但我的问题的重点是知道它是否可以用 std::transform 或其他算法来完成?

son*_*yao 4

您可以添加 lambda 捕获来存储前一个元素的值。

例如

std::vector<int> v2(v1.size());
std::transform(v1.begin(), v1.end(), v2.begin(),
    [pv = 0](int val) mutable { int nv = pv + val; pv = val; return nv; });
Run Code Online (Sandbox Code Playgroud)

居住

PS:自 C++14 起支持使用初始化程序进行捕获。