[프로그래머스][1단계][python] 약수의 합 문제

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

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

입출력 예

n	return
12	28
5	6

입출력 예 설명
- 입출력 예 #1
12의 약수는 1, 2, 3, 4, 6, 12입니다. 이를 모두 더하면 28입니다.

- 입출력 예 #2
5의 약수는 1, 5입니다. 이를 모두 더하면 6입니다.

 

def solution(n):

    arr = list()
    # 1부터 n까지 나눠보고..
    for i in range(1,n+1):
        if n%i==0: # 나누어떨어지면 약수리스트에 삽입
            arr.append(i)

    return sum(arr)

>>> 리스트 컴프리헨션 사용하면 더 간단해짐

def solution(n):

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