C++ std::accumulate 간단 사용법 및 주의사항

알고리즘 문제 풀 때 accumulate 함수를 많이 사용하실텐데요, 이 함수로 간편하게 배열의 합을 구할 수 있습니다.

이 함수는 <numeric> 헤더에 들어있습니다.

 

 

#include <iostream>
#include <vector>
#include <numeric>
#include <string>
#include <functional>
 
int main()
{
    std::vector<int> v{1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
 
    int sum = std::accumulate(v.begin(), v.end(), 0);
 
    int product = std::accumulate(v.begin(), v.end(), 1, std::multiplies<int>());
 
    auto dash_fold = [](std::string a, int b) {
                         return std::move(a) + '-' + std::to_string(b);
                     };
 
    std::string s = std::accumulate(std::next(v.begin()), v.end(),
                                    std::to_string(v[0]), // start with first element
                                    dash_fold);
 
    // Right fold using reverse iterators
    std::string rs = std::accumulate(std::next(v.rbegin()), v.rend(),
                                     std::to_string(v.back()), // start with last element
                                     dash_fold);
 
    std::cout << "sum: " << sum << '\n'
              << "product: " << product << '\n'
              << "dash-separated string: " << s << '\n'
              << "dash-separated string (right-folded): " << rs << '\n';
}

sum: 55
product: 3628800
dash-separated string: 1-2-3-4-5-6-7-8-9-10
dash-separated string (right-folded): 10-9-8-7-6-5-4-3-2-1

cppreference에서 가져온 예시입니다.

여러가지 예시중에 Binary Operation을 사용한 것들은 참고로 보시고, 단순히 합을 구하는 것에 대해서만 써보겠습니다.

 

 

vector<int> v = { 1, 2, 3, 4, 5 };
cout << accumulate(v.begin(), v.end(), 0) << '\n';

//출력: 15 = 1 + 2 + 3 + 4 + 5

accumulate의 첫번째, 두번째 인자는 first, last iterator가 들어갑니다. 세번째 인자는 'initial value of the sum', 즉 합의 초기값입니다.

 

 

vector<int> v = { 1'000'000'000, 2'000'000'000 };
long long sum = accumulate(v.begin(), v.end(), 0);
cout << sum << '\n';	//출력: -1294967296

주의해야 할 점은 범위를 초과하는 값을 더할 때 초기값을 잘 설정해줘야 한다는 겁니다.

보시면 int형 범위를 초과하는 값을 더하기 위해 long long sum이라는 변수를 만들었는데, 출력을 해보면 오버플로우가 발생됩니다.

이 문제는 accumulate의 반환값이 initial value의 타입을 따라가기 때문입니다. int값인 0을 넣었으니 accumulate의 반환값도 int로 정해지게 됩니다.

warning도 뜨지 않고 자동으로 캐스팅되기 때문에 주의하셔야 합니다.

 

 

template< class InputIt, class T >
constexpr T accumulate( InputIt first, InputIt last, T init );

함수 원형을 보시면 템플릿으로 반환값이 init과 동일하다는걸 알 수 있습니다.

 

 

vector<int> v = { 1'000'000'000, 2'000'000'000 };
long long sum = accumulate(v.begin(), v.end(), 0LL);
cout << sum << '\n';	//출력: 3'000'000'000

long long 타입을 쓸 생각이라면 initial value에 0 대신 0LL을 넣어줘야 합니다.

여러분은 PS하실때 저같은 실수 하지 않기를 바랍니다..