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

문제 설명

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

 

 

내 코드

def solution(left, right):
    total = 0 # 총합을 담아줄 변수
    for i in range(left, right+1): # left 부터 right까지 반복 
        divisors = set() # 약수를 중복없이 담아줄 set 객체 생성
        for j in range(1, i+1): # i의 약수개수 구하기
            if i % j == 0:
                divisors.add(j) 
                divisors.add(i // j) # 약수는 n = a * b이므로 i를 j로 나누면 다른 약수를 찾을 수 있음
        if len(divisors) % 2 == 0: # 해당 수의 약수 개수가 짝수인지 확인
            total += i
        else:
            total -= i
    return total

# left부터 right 범위의 수들의 약수를 구함
# 약수의 개수가 짝수라면 결과에 더해줌 홀수라면 빼줌
# 약수는 n = a*b임. 즉 a를 찾으면 b도 찾을 수 있음

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/77884

프로그래머스