[Level 1 / Python] 약수의 합

문제 설명

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

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

출처 : https://programmers.co.kr/learn/courses/30/lessons/12928

풀이 방법

리스트 컴프리헨션을 이용하여 입력 받은 n보다 작은 자연수로 나누었을 때 나누어 떨어지는 숫자만을 리스트로 만들었습니다.

소스 코드

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