[Swift Algoritm] 평균 구하기 Programmers(Lv.1)

문제 설명

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

 

제한 조건

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

 

입출력 예

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

 

내 문제 풀이

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

• reduce(0, +)를 이용해서 arr 배열의 모든 원소를 더해주고, arr.count로 구한 배열의 길이로 나누어주었다.

 

💡 피드백

• 굳이 result 상수를 선언하지 않고 바로 return문 한 줄에서 끝내도 됐을 듯 싶다.
• reduce 고차 함수의 정확한 의미 및 사용 방법에 대한 공부 필요!! 다음 글에서 다뤄야겠다.

 

---

 

문제

https://programmers.co.kr/learn/courses/30/lessons/12944