본문 바로가기

알고리즘/Python 기초

파이썬 - 약수의 합(lv.1)

문제 ) 프로그래머스 - 약수의 합(lv.1)

📌 문제 설명

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

 

📌 제한 사항

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

 

📌 입출력 예 

n return
12 28
5 6

✏️ for 반복문, if 조건문

def solution(n):
    answer = 0
    for i in range(1, n + 1):
        if n % i == 0:
            answer += i
    return answer
    
    # 한 줄로 바꿔보기
    return sum([i for i in range(1, n + 1) if n % i == 0])