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

시작하며...

모든 알고리즘 문제 풀이는 제가 직접 짜서 정답을 맞춘 결과만을 공유합니다.

마지막 'More Solution'은 다른 정답자들 풀이 중 생각지 못했던 부분들이나 좋게 느껴진 풀이법 몇개를 가져와서 공유하였습니다.

 

[Level 1] 약수의 합

프로그래머스

 

문제 설명

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

제한 사항

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

입출력 예

입출력 예 설명

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

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

 

My Solution

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

약수들의 합을 구해주기 위해서 반복문을 돌면서 나머지가 0인 경우에 answer 변수에 합쳐주는 작업을 했다.

성능 향상을 위해 n까지 전체 반복문을 도는것이 아닌 n의 제곱근까지만 돌게 구성했으며 그랬을 때 나눈 값과 몫을 같이 합쳐줘야한다. 

그리고 n이 어떤수의 제곱인 숫자일 경우에는 두번 더해주는 이슈가 있어서 한번 더 조건문을 추가했다.

성능향상을 위해 제곱근을 썼는데 if문이 두번인 것이 아쉽다...좀 더 나은 풀이가 있을것 같은데.

 

 

More Solution

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

List Comprehension으로 한줄로 깔끔하게 구성된 코드.

이 경우는 range 범위가 num // 0.5이 되면 답이 다르게 나올것이고 num // 2가 되어주어야한다.

 

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges