약수의 개수와 덧셈
문제 설명
두 정수 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 해야 합니다.
===============================================
내가 작성한 코드
import Foundation
func solution(_ left:Int, _ right:Int) -> Int {
let tempList = [Int](left...right)
var result: Int = 0
result = tempList.map { (num: Int)->Int in
var count: Int = 0
for i in 1...num {
if num % i == 0{
count += 1
}
}
if count % 2 == 0 {
return num
} else {
return -num
}
}.reduce(0, +)
return result
}
이번 코드를 작성할 때는 고차함수를 무조건 사용해야 겠다 싶어서 이렇게 작성했다. 아니 근데 세상에 나는 filter를 생각하지 못했다. 다른 사람 풀이를 보는데 맨 위의 분의 코드는 정말 깔끔 그 자체였다.
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, +)
}
이 두개의 솔루션의 성능 차이는 어마무시했다.
import Foundation
func solution(_ left:Int, _ right:Int) -> Int {
let tempList = [Int](left...right)
var result: Int = 0
result = tempList.map { (num: Int)->Int in
var count: Int = 0
for i in 1...num {
if num % i == 0{
count += 1
}
}
if count % 2 == 0 {
return num
} else {
return -num
}
}.reduce(0, +)
return result
}
func solution2(_ left: Int, _ right: Int) -> Int {
return (left...right).map { i in (1...i).filter { i % $0 == 0 }.count % 2 == 0 ? i : -i }.reduce(0, +)
}
var start = CFAbsoluteTimeGetCurrent()
print(solution(13, 17))
var processTime = CFAbsoluteTimeGetCurrent() - start
print("경과 시간: \(processTime)")
start = CFAbsoluteTimeGetCurrent()
print(solution2(13, 17))
processTime = CFAbsoluteTimeGetCurrent() - start
print("경과 시간: \(processTime)")
결과
43
경과 시간: 0.006428956985473633
43
경과 시간: 0.0008729696273803711
약 7배~8배의 차이가 났다. 역시 고차 함수를 많이 사용할 수록 좋은 성능이나오는것을 다시 한번 깨닳았다.