[프로그래머스 : PYTHON] 약수의 개수와 덧셈

문제 설명

두 정수 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 해야 합니다.

입출력 예 #2

• 다음 표는 24부터 27까지의 수들의 약수를 모두 나타낸 것입니다.

수	약수	약수의 개수
24	1, 2, 3, 4, 6, 8, 12, 24	8
25	1, 5, 25	3
26	1, 2, 13, 26	4
27	1, 3, 9, 27	4

• 따라서, 24 - 25 + 26 + 27 = 52를 return 해야 합니다.

 

 

문제풀이

나의 풀이는 left ~ right 범위의 수를 하나씩 돌면서 해당 숫자의 제곱수 만큼의 숫자들이 약수가 몇 개인지 판별을 했다. 내가 이중 for문으로 풀었다면 다른 사람의 풀이는 굉장히 간단했다.

 

약수가 홀수개라면 제곱근이 정수였을 것이지만 짝수개라면 그렇지 않기 때문에 이 포인트에 집중해서 푼다면 하나의 for문으로 문제 해결이 가능하다.

 

전체코드

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

    return answer

 

다른 사람 코드

import math
def solution(left, right):
    answer = 0
    
    for i in range(left, right+1):
        if int(math.sqrt(i))  == math.sqrt(i):
            answer -= i
        else:
            answer += i
    
    return answer