[Swift] lv.1 약수의 개수와 덧셈 | Programmers

문제 설명

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

제한 사항

1 ≤ left ≤ right ≤ 1,000

1차 고민

이중for문이 맞다고 느껴지긴한다. 더 많은 고차함수를 사용할 수 있었겠지만 가장 머릿속에서 빠르게 떠오른 방법은 i j를 이용한 for문의 중첩이라...

1차 코드

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
            }
        }

        if count % 2 == 0 {
            result += i
        } else {
            result -= i
        }
    }
    return result
}

1차 결과

다른 사람의 풀이

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

map을 통한 left부터 right까지 배열화.

filter를 통해 i의 약수를 count해서 홀,짝에 대한 삼항연산자. 그에 대한 참과 거짓을 reduce로 중첩.