프로그래머스 - 약수의 합 / Python

문제: https://programmers.co.kr/learn/courses/30/lessons/12928#qna

코딩테스트 연습 - 약수의 합

문제 설명

문제 설명

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

제한 사항

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

입출력 예

nreturn

12	28
5	6

입출력 예 설명

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

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

 

---

정답

def solution(n):
    answer = 0
    for i in range(1, int(n ** 0.5) + 1):
        if n % i == 0:
            answer += i
            if i != n // i:
                answer += n // i
    return answer

 

---

풀이

 n의 약수의 합을 구하는 문제입니다. 에라토스테네스의 체처럼 n의 제곱근까지만 확인하고, 나눠지는 경우 나누는 값과 몫을 더해주면 됩니다. 

 

제곱근의 경우에는 약수가 두 번 더해지기 때문에 if문으로 걸러주는 것이 중요합니다.