[Swift｜코드카타] (프로그래머스) 입문 #10. 배열의 평균값 - reduce 함수

문제 (링크)

• 문제 : 정수 배열 numbers가 매개변수로 주어집니다. numbers의 원소의 평균값을 return하도록 solution 함수를 완성해주세요.
• 제한사항 :
  • 0 ≤ numbers의 원소 ≤ 1,000
  • 1 ≤ numbers의 길이 ≤ 100
  • 정답의 소수 부분이 .0 또는 .5인 경우만 입력으로 주어집니다.

 

​풀이 1 > for...in 구문 사용

import Foundation

func solution(_ numbers:[Int]) -> Double {
    guard numbers.count <= 100 else {
        print("numbers의 길이는 100 이하입니다.")
        return 0
    }
    var sum = 0
    var average: Double = 0
    for i in numbers {
        sum += i
    }
    average = Double(sum) / Double(numbers.count)
    return average
}

 

풀이 2 > ✨reduce✨ 사용

import Foundation

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

 

reduce를 사용하면 코드가 훨씬 깔끔해진다.

 

reduce 함수에 대한 자세한 내용은 다음 글에서 확인 가능하다.

2024.02.07 - [Swift/TIL : Today I Learned] - [Swift] 고차함수 - reduce

[Swift] 고차함수 - reduce