[Swift｜코드카타] (프로그래머스) 입문 #12. 평균 구하기

🔴 문제

• 문제 : 정수를 담고 있는 배열 arr의 평균값을 return하는 함수, solution을 완성해보세요.
• 제한사항 :
  • arr은 길이 1 이상, 100 이하인 배열입니다.
  • arr의 원소는 -10,000 이상 10,000 이하인 정수입니다.

 

🔵 풀이 1 > for...in 사용

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

 

 

🔵 풀이 2 > reduce 사용

func solution(_ arr:[Int]) -> Double {
    return Double(arr.reduce(0,+)) / Double(arr.count)
}

 

 

※ reduce 함수 설명 : https://yurim-dev.tistory.com/16

[Swift] 고차함수 - reduce

 

 

 

🟢 코드 리뷰

코드 간결화를 위해 reduce 함수 사용에 익숙해져야겠다.