[swift] 프로그래머스 - 소인수분해

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.

 

고차함수를 사용하지 않은 풀이

import Foundation

func solution(_ n:Int) -> [Int] {
    var num = n
    var result: Set<Int> = []
    var count = 2
    while num != 1 {
        if num % count == 0 {
            num = num / count
            result.update(with: count)
            print(result)
            count = 2
            print(num)
        } else {
            count += 1
        }
    }
    return Array(result).sorted()
}

•  

 

개선한 풀이

import Foundation

func solution(_ n:Int) -> [Int] {
    var result: [Int] = []
    for i in (2...n) {
        if n % i == 0 && !result.contains(where: { i % $0 == 0}) {
            result.append(i)
        }
    }
    return result
}

• 코드길이 감소.