[프로그래머스] 평균 구하기(C++)

문제 설명

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

 

제한사항

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

입출력 예

arr                                                                                                     return

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

 

 

기본적인 문제다.

그나마 주의할점이 있다면 자료형 ? 정도 라고 생각한다.

 

#include <iostream>
#include <string>
#include <vector>

using namespace std;

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

    for(int i = 0; i < arr.size(); i++)
    {
        sum = sum + arr[i];
    }

    answer = sum / arr.size();
    return answer;
}

//테스트 실행용
int main(void)
{
    vector<int> arr = {1,2,3,4};
    cout << solution(arr);
}