[프로그래머스] 평균 구하기 c++

문제 설명

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

제한 사항

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

입출력 예

arr	return
[1,2,3,4]	2.5
[5,5]	5

풀이

sum 기능을 하는 accumulate 함수를 사용해서 풀었다.

#include <string>
#include <vector>
#include <numeric>

using namespace std;

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