[C++] 벡터 요소의 합 구하기 accumulate

📌 import 헤더파일

#include <numeric> // accumulate를 사용하기 위해 필요

 

<numeric>는 C++ 표준 라이브러리의 헤더 파일 중 하나로,  벡터나 배열 등의 데이터 집합에 대해 누적 합, 내적, 차분 계산 등을 간단하게 수행할 수 있는 유용한 함수들을 제공한다. 

 

코딩 테스트 문제를 풀 때 벡터나 배열 요소를 빠르고 간편하게 누적할 수 있다.

 

📌 인자 설명

✅ 첫번째 인자 : 시작 반복자(범위의 시작)

✅ 두번째 인자 : 끝 반복자(범위의 끝)

✅ 세번째 인자: 누적 연산의 초기값

 

📌 예시

✅ 기본적인 벡터 요소의 합

#include <numeric> // accumulate를 사용하기 위해 필요
#include <vector>
#include <iostream>
using namespace std;

vector<int> vec = {1, 2, 3, 4, 5};

int main() {
   
    int sum = accumulate(vec.begin(), vec.end(), 0); // 0은 초기값
    
    cout << "벡터 요소의 합: " << sum << "\n";
    
    return 0;
}

 

✅ 초기값을 활용한 누적

👉🏻 초기값 100으로 설정 시

#include <numeric>
#include <vector>
#include <iostream>
using namespace std;

vector<int> numbers = {1, 2, 3, 4};

int main() {
  
    int sum_with_initial = accumulate(numbers.begin(), numbers.end(), 100);
    
    cout << "초기값 100을 포함한 합: " << sum_with_initial << "\n"; // 출력: 110
    
    return 0;
}

 

✅ 사용자 정의 함수로 곱, 차, 조건부 등의 계산도 가능하다.

accumulate의 네번째 인수로 사용자 정의 연산을 전달할 수 있다.

#include <numeric>
#include <vector>
#include <iostream>
using namespace std;

vector<int> numbers = {1, 2, 3, 4};

int main() {
    int product = accumulate(numbers.begin(), numbers.end(), 1, [](int a, int b) {
        return a * b;
    });
    
    cout << "벡터 요소의 곱: " << product << "\n"; // 출력: 24


    return 0;
}

#include <numeric>
#include <vector>
#include <iostream>
using namespace std;

//초기값 100으로 잡고 빼기

vector<int> numbers = {1, 2, 3, 4};

int main() {
    int product = accumulate(numbers.begin(), numbers.end(), 100, [](int a, int b) {
        return a - b;
    });
    
    cout << "벡터 요소의 차: " << product << "\n"; // 출력: 24


    return 0;
}

//벡터 요소의 차: 90

#include <numeric>
#include <vector>
#include <iostream>
using namespace std;

vector<int> numbers = {1, 2, 3, 4, 5, 6};

int main() {
    int even_sum = std::accumulate(numbers.begin(), numbers.end(), 0, [](int a, int b) {
        return b % 2 == 0 ? a + b : a;
    });

    cout << "짝수의 합: " << even_sum << "\n"; // 출력: 12
    
    return 0;
}

 

✅ 숫자뿐만 아니라 문자열 연산도 가능하다.

#include <numeric>
#include <vector>
#include <string>
#include <iostream>
using namespace std;

vector<string> words = {"Hello", " ", "World", "!"};

int main() {
    
    string sentence = accumulate(words.begin(), words.end(), string(""));
    cout << "문장: " << sentence << "\n"; // 출력: Hello World!
    return 0;
}