[Swift] 알고리즘 공부 - 약수의 합

문제 설명

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

 

 

제한 사항

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

 

입출력 예

 

n                     return

12	28
5	6

 

설명

1. 1부터 n까지 나눠서 나머지가 0이면 저장시킨다.

2. 0은 예외처리 해준다.

 

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

    return answer
}

 

0을 예외처리 해주는것만 생각하면 쉬운문제다