[Swift] 알고리즘 공부 - 평균 구하기

문제 설명

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

 

제한사항

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

 

입출력 예

arr                             return

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

 

설명

1. arr의 모든수를 더한다

2. 모든수를 더한값에서 배열에 수만큼 나눠준다

func solution(_ arr:[Int]) -> Double {
    var sum: Int = 0
    
    for i in arr {
        sum += i
    }
    
    return Double(sum) / Double(arr.count)
}

 

Double은 계산하려는 값이 모두 Double이어야지 연산이 가능하기하다