[프로그래머스 Level 1] 평균 구하기

문제

평균 구하기  

 

정수를 담고 있는 배열 arr의 평균값을 return하는 함수, solution을 완성해보세요.

 

제한사항

1. arr은 길이 1 이상, 100 이하인 배열입니다.
2. arr의 원소는 -10,000 이상 10,000 이하인 정수입니다.

 

코드

#include <vector>

using namespace std;

double solution(vector<int> arr)
{
    double answer = 0;

    for (int i = 0; i < arr.size(); i++)
        answer += arr[i];

    return answer / arr.size();
}

 

나의 생각

numeric 라이브러리에 accumulate 함수가 있다는 걸 배웠다.

 

accumulate는 [first, last) 범위의 원소를 더해나간다. 세 번째 매개변수는 초깃값이다. 

std::accumulate 레퍼런스

#include <vector>
#include <numeric>

using namespace std;

double solution(vector<int> arr)
{
    double answer = accumulate(arr.begin(), arr.end(), 0);
    return answer / arr.size();
}