[Swift] lv.1 평균 구하기 | Programmers

문제 설명

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

제한 사항

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

1차 코드

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

    var sum = 0

    for i in 0..<arr.count {
        sum += arr[i]
    }

    var avg = Double(sum) / Double(arr.count)

    return avg
}

1차 결과

미흡했던 점

Double형으로 바꾸지 않고 그냥 평균을 내버릴 뻔 했다. 소수점이 삭제되는 문제가 생길 수 있었다.