在 for_each 上使用函子

Question

为什么函子上的 for_each 调用最后没有更新 sum::total ？

struct sum
{
    sum():total(0){};
    int total;

    void operator()(int element) 
    { 
       total+=element; 
    }
};

int main()
{
    sum s;

    int arr[] = {0, 1, 2, 3, 4, 5};
    std::for_each(arr, arr+6, s);
    cout << s.total << endl; // prints total = 0;
}

Answer 1

for_each 按值获取函子 - 因此它被复制。例如，您可以使用一个用指向外部 int 的指针初始化的函子。

struct sum
{
    sum(int * t):total(t){};
    int * total;

    void operator()(int element)
    {
       *total+=element;
    }
};

int main()
{
    int total = 0;
    sum s(&total);

    int arr[] = {0, 1, 2, 3, 4, 5};
    std::for_each(arr, arr+6, s);
    cout << total << endl; // prints total = 15;
}

或者您可以使用 for_each 的返回值

struct sum
{
    sum():total(0){};
    int total;

    void operator()(int element) 
    { 
       total+=element; 
    }
};

int main()
{
    sum s;

    int arr[] = {0, 1, 2, 3, 4, 5};
    s = std::for_each(arr, arr+6, s);
    cout << s.total << endl; // prints total = 15;
}

Answer 2

for_each 按值接收函子的副本。即使在那之后，它也可以自由复制它，但会返回一个副本。

OTOH，您只是想重新发明 std::accumulate ，这会更轻松地完成这项工作：

int total = std::accumulate(arr, arr+6, 0);
cout << total << endl; 

Answer 3

因为您传递给 for_each 的 s 是按值传递的。 for_each 按值接受它！

在 C++0x 中，您可以使用 for_each 解决此问题，

int sum  = 0;
std::for_each(arr, arr+6, [&](int n){ sum += n; });
std::cout << sum ;

输出：

15

IDEONE 上的演示：http ://ideone.com/s7OOn

---

或者您可以简单地在 std::cout 本身中编写：

std::cout<<std::for_each(arr,arr+6,[&](int n)->int{sum += n;return sum;})(0);

Run : http://ideone.com/7Hyla

请注意，这种不同的语法对于学习目的是可以的，例如 std::for_each 如何工作以及它返回什么，但是我不会在实际代码中推荐这种语法。 :-)

---

在 C++ 中，您可以在仿函数中编写用户定义的转换函数，

struct add
{
    int total;
    add():total(0){};
    void operator()(int element)  {  total+=element;  }
    operator int() { return total ; }
};

int main()
{
    int arr[] = {0, 1, 2, 3, 4, 5};
    int sum = std::for_each(arr, arr+6, add());
    std::cout << sum;
}

它与 Erik 第二个解决方案的版本略有不同： http ://ideone.com/vKnmA

Answer 4

发生这种情况是因为 std::for_each 要求函子按值传递。
针对您的解决方案的解决方法：

struct sum
{
    sum():total(0){};
    int total;
    sum(sum & temp)
    {
        total = temp.total;
    }
    void operator()(int element) 
    { 
       total+=element; 
    }
};

int main()
{
    sum s;

    int arr[] = {0, 1, 2, 3, 4, 5};
    s = std::for_each(arr, arr+6, s);  // result of for_each assigned back to s
    cout << s.total << endl; // prints total = 0;
}

Answer 5

如果您希望对象状态在 std::for_each 之后更新，std::ref() 也是另一个选择

struct Sum
{
    int total = 0;
    void operator()(int i) { total += i; }
};

int main()
{
    int arr[] = { 0, 1, 2, 3, 4, 5 };

    Sum obj1;
    Sum t1 = std::for_each(arr, arr + 6, obj1);  // t1.total = 15 
                            // (Note: obj1.total = 0 bcos it is passed by value)

    Sum obj2;
    std::for_each(arr, arr + 6, std::ref(obj2)); // obj2.total = 15 (NO copy)

    Sum t2 = std::for_each(arr, arr + 6, Sum()); // t2.total = 15
}