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

문제 설명

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

제한 조건

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

예시

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

Solution.Swift

func solution(_ arr:[Int]) -> Double {
    var sum: Int = 0

    for i in arr {
        sum += i
    }

    return Double(sum) / Double(arr.count)
}

1. 합을 구할 sum 변수 생성
2. for문으로 sum에 변수의 총합 저장
3. 총합의 평균을 구해 return

---

다른 사람의 문제풀이

func solution(_ arr:[Int]) -> Double {

    return Double(arr.reduce(0,+))/Double(arr.count)
}

reduce를 활용하여 깔끔하게 처리함

[프로그래머스 테스트]