프로그래머스 약수의 개수와 덧셈 (파이썬)

문제 설명

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

---

제한사항

• 1 ≤ left ≤ right ≤ 1,000

---

입출력 예

leftrightresult

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):
    answer = 0
    arr = []
    for i in range(1001):
      arr.append(0)
    for a in range(1, 1001):
      for b in range(0, 1001, a):
        arr[b] += 1
    for i in range(left, right+1):
      if arr[i]%2 == 0:
        answer += i
      else:
        answer -= i
    return answer

다이나믹 프로그래밍인줄 알고 DP로 접근하였더니 다른사람의 풀이를 보니

제곱수의 약수는 무조건 홀수이므로 루트를 씌워 판단하였다

루트를 이용한 풀이가 메모리 + 속도 모두 효율성이 좋은 풀이다

본인의 코드는 속도는 O(1)로 접근 가능하나 메모리를 많이잡아먹는다