연습문제 > 평균 구하기

문제 설명

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

제한사항

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

입출력 예

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

 

Solution

#include <string>
#include <vector>

using namespace std;

double solution(vector<int> arr) {
  double answer = 0;
  for (auto x : arr) {
    answer += x;
  }

  return answer / arr.size();
}

 

남의 코드

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

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

    return answer / arr.size();
}

 

후기

이런 문제를 남들은 어떻게 푸는 지 궁금했는데 STL을 이용하는구나.

 

(url: https://programmers.co.kr/learn/courses/30/lessons/12944)