알고리즘
[Swift 알고리즘] - 약수의 합
Peppo
2022. 5. 4. 10:21
728x90
https://programmers.co.kr/learn/courses/30/lessons/12928
코딩테스트 연습 - 약수의 합
정수 n을 입력받아 n의 약수를 모두 더한 값을 리턴하는 함수, solution을 완성해주세요. 제한 사항 n은 0 이상 3000이하인 정수입니다. 입출력 예 n return 12 28 5 6 입출력 예 설명 입출력 예 #1 12의 약수
programmers.co.kr
내 풀이
func solution(_ n:Int) -> Int {
// TODO: [x] 약수들을 담을 배열
var aliquotArr: [Int] = []
// TODO: [x] 약수 구하기
for num1 in 0...n {
for num2 in 0...n {
if num1 * num2 == n {
aliquotArr.append(num1)
}
}
}
// TODO: [x] 약수 배열내부의 전체 합
return aliquotArr.reduce(0) { $0 + $1 }
}
solution(12)
다른사람의 풀이
func solution(_ n:Int) -> Int {
return n != 0 ? (1...n).filter{n % $0 == 0}.reduce(0){$0 + $1} : 0
}
1. 정수가 0이 아니면 아래 2~3번 실행 / 0이면 return 0
2. 1부터 n까지의 수를 이용해 나머지값이 0이 되는것만 걸러낸다.(filter)
3. 걸러져 나온 배열의 요소들을 다 합한다.
배운것
reduce(_:_:)
컨테이너의 요소들을 결합한 결과를 반환합니다.
reduce는 여기에 자세히 정리 해놓았습니다.
728x90