[프로그래머스] 약수의 합 (LV.1)

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

 

제한 사항
n은 0 이상 3000이하인 정수입니다.

 

입출력 예

def solution(n):
    answer = []
    for i in range(n):
        if n%i == 0:
                answer.append(i)
    return sum(answer)

ZeroDivisionError: integer division or modulo by zero

라는 에러가 떴다. 위 에러는 숫자를 0으로 나누었기 때문이다.

range 안에 0이 아닌 1부터 n+1(마지막 자기자신인 n값까지) 를 정의해주니 해결되었다.

def solution(n):
    answer = []
    i = 1
    for i in range(1, n+1):
        if n%i == 0:
                answer.append(i)
    return sum(answer)

 

최적 답안: for문을 한 줄에 써서 최종적으로 한 줄에 다 넣었음

def sumDivisor(num):
    # num / 2 의 수들만 검사하면 성능 약 2배 향상잼
    return num + sum([i for i in range(1, (num // 2) + 1) if num % i == 0])