[프로그래머스/python3] 약수의 개수와 덧셈

문제 설명

두 정수 left와 right가 매개변수로 주어집니다. left부터 right까지의 모든 수들 중에서, 약수의 개수가 짝수인 수는 더하고, 약수의 개수가 홀수인 수는 뺀 수를 return 하도록 solution 함수를 완성해주세요.

 

제한 사항

• 1 ≤ left ≤ right ≤ 1,000

 

입출력 예

 

left	right	result
13	17	43
24	27	52

 

입출력 예 설명

#1) 다음 표는 13부터 17까지의 수들의 약수를 모두 나타낸 것입니다.

 

수	약수	약수의 개수
13	1, 13	2
14	1, 2, 7, 14	4
15	1, 3, 5, 15	4
16	1, 2, 4, 8, 16	5
17	1, 17	2

• 따라서, 13 + 14 + 15 - 16 + 17 = 43을 return 해야 합니다.

---

def solution(left, right):
    answer = 0
    count = 0
    for i in range(left, right+1):
        for j in range(i):
            if 0 == i %(j+1):
                count += 1
        if count %2 == 0:
            answer += i
        else:
            answer -= i
        count = 0

    return answer

라고 초딩처럼 풀었는데 약수가 홀수개인 수는 모두 제곱수라고 한다.

이를 이용하여 제곱근이면 - 제곱근이 아니면 +를 붙여주는 방식으로 답을 구할 수 있으므로 훨씬 쉬워진다.

 

이를 이용한 풀이는 아래와 같다.

import math

#solution1
#math.sqrt(x) 함수는 x의 제곱근을 반환한다
def solution(left, right):
    answer = 0
    for i in range(left, right + 1, 1):
        sqrt = math.sqrt(i)
        if int(sqrt) == sqrt:
            answer -= i
        else:
            answer += i

    return answer

#solution2
def solution(left, right):
    answer = 0
    for i in range(left,right+1):
        if int(i**0.5)==i**0.5:
            answer -= i
        else:
            answer += i
    return answer