[iOS Swift] Lv.1 평균 구하기

출처 : 프로그래머스
https://school.programmers.co.kr/learn/courses/30/lessons/12944
난이도 : Lv.1

 

문제 설명

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

 

제한 사항

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

입출력 예

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

---

func solution(_ arr:[Int]) -> Double {
    var sum: Int = 0 // sum값을 정수형 0으로 초기화
    
    for i in 0..<arr.count{ //arr의 길이값 만큼 실행
        sum += arr[i]   // 배열 arr의 값의 합을 sum에 저장
    }
    print(sum)  // 제대로 정보가 들어갔는지 확인
    
    return ((Double(sum)/Double(arr.count))) //평균값을 값을 리턴
}