프로그래머스 level 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):
    a = []
    for i in range(1, n+1):
        if n % i == 0:
            a.append(i)
   
    return sum(a)

위의 코드를 한 줄로 나타낼 수 있다.

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