프로그래머스 약수의 합 Swift

문제 설명

정수 n을 입력받아 n의 약수를 모두 더한 값을 리턴하는 함수, solution을 완성해주세요.

제한 사항

• n은 0 이상 3000이하인 정수입니다.

입출력 예

n	return
5	6
12	28

풀이:걍 0일 때만 조심하고 나머지는 for문 돌려서 나누어 떨어지는 것들을 더해주면 됨.

 

func solution(_ n:Int) -> Int {
    var count = 0
    for i in 1...n{
        if n%i == 0 {
            count += i
        }
    }
    return n == 0 ? 0 :count
}

다른 사람 풀이 중 가장 좋다고 생각하는 것

우선 guard문을 이용해 0일 경우를 대비하고

Array와 filter를 이용해 나누어 떨이지는 것을 넣고 reduce를 이용해서 차례로 더했다.

나도 reduce를 생각했는데 대충 머리쓰기 귀찮아서 for문을 돌렸다.

func solution(_ n:Int) -> Int {   
    guard n != 0 else {
        return 0
    }
    return Array(1...n).filter{n % $0 == 0}.reduce(0, +)
}