약수의 합
문제 설명
정수 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):
sum = 0
for i in range(1, (n // 2) + 1):
if n % i == 0:
sum += i
return sum + n
정수 n의 진약수는 n의 절반 + 1부터는 존재하지 않습니다.
그러므로 1부터 n의 절반까지 반복문을 돌면서 약수를 계속해서 누적해주고,
마지막에 자기 자신인 n을 더해주면 n의 약수를 모두 더한 값을 구할 수 있습니다.
두 번째 풀이
def solution(n):
return sum([i for i in range(1, (n // 2) + 1) if n % i == 0]) + n
첫 번째 풀이 방법과 원리는 동일하지만,
파이썬의 List Comprehension 문법을 사용하면 라인 수를 줄일 수 있습니다.
List Comprehension으로 진약수의 리스트를 만든 후 sum() 함수로 진약수를 모두 더한 값을 구합니다.
그리고 진약수를 모두 더한 값에 자기 자신인 n을 더해주면 n의 약수를 모두 더한 값이 나오게 됩니다.