약수의 개수와 덧셈

문제

• 약수의 개수와 덧셈

Description

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

 

풀이

import Foundation

func solution(_ left: Int, _ right: Int) -> Int {
    return (left...right).map { i in (1...i).filter { i % $0 == 0 }.count % 2 == 0 ? i : -i }.reduce(0, +)
}

 

해석

1. (left...right).map { i in ... }: 매개변수 left에서 right까지의 범위에서 mapping i가 1과 i 사이에 있을 때
2. (1...i).filter { i % $0 == 0 }.count: 현재의 수 i에 대해 약수를 찾기 위해 1부터 i까지의 수 중에서 i로 나누어 떨어지는 수를 찾고, 그 개수를 세줍니다. 이때 filter 함수를 사용하여 조건에 맞는 수들을 걸러내고, count를 통해 개수를 계산합니다.
3. .count % 2 == 0 ? i : -i: 약수의 개수가 짝수인지 홀수인지를 판단합니다. 짝수인 경우에는 현재의 수 i를 반환하고, 홀수인 경우에는 -i를 반환합니다. 이렇게 함으로써 짝수인 경우에는 더하고, 홀수인 경우에는 빼게 됩니다.
4. .reduce(0, +): 각 수에 대해 계산된 값을 모두 더하여 최종 결과를 반환합니다.