[Swift Algorithm] 약수의 개수와 덧셈 Programmers(Lv.1)

문제 설명

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

 

제한 조건

• 1 ≤ left ≤ right ≤ 1,000

 

입출력 예

left	right	result
13	17	43
24	27	52

 

내 문제 풀이

import Foundation

func solution(_ left:Int, _ right:Int) -> Int {
    var result = 0
    for i in left...right {
        var count = 0
        for j in 1...i {
            if i % j == 0 { count += 1 }
        }
        count % 2 == 0 ? (result += i) : (result -= i)
    }
    return result
}

• 어차피 left는 right보다 작거나 같으므로 따로 예외처리 해 줄 필요 없이, 바로 left...right로 반복문을 작성했다.
• left부터 right까지 모든 수를 1부터 해당 수까지로 나눠보며 약수의 개수를 구해주었다.
• 약수의 개수가 짝수면 더해주고, 홀수면 빼주는 것을 반복해주어 누적된 result 값을 반환해주었다.

 

인상적인 풀이

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, +)
}

• 가독성 측면에서 보면 좋은 코드인지는 잘 모르겠지만, 고차 함수 사용 부분에서 배울점이 많아 가져와봤다.
• 우선 (left...right).map 처럼 바로 범위에 map을 사용할 수 있다는 점이 굉장히 신기했다,,
• 한 줄에 map, filter, reduce가 모두 들어간 코드,, 역시 활용할 점이 많다.

 

💡 피드백

• 고차함수를 적절히 잘 섞어 작성했다면 더 간결하게 코드를 완성할 수 있었을 것 같다.

 

---

 

문제

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