약수의 개수와 덧셈

[문제]

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

 

[풀이]

이 문제는 어떤 정수의 약수의 개수가 홀수인지 짝수인지 판별하는 방법만 알면 효율적으로 풀 수 있다.

약수는 쌍으로 존재하는데, 제곱수의 경우 그 쌍이 자기 자신이다. 즉, 제곱수의 약수의 개수는 홀수개이다.

이를 이용해 문제를 풀어보자

 

def solution(left, right):
    
    return sum(-n if (n**0.5).is_integer() else n 
               for n in range(left, right+1))