[프로그래머스] 약수의 합 - python

약수의 합

문제 설명

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

제한사항

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

풀이

절반까지만 비교하면 중복을 제거할 수 있다. 단 n을 더해줘야한다.

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