[Swift] 코딩테스트 연습! Lv0. 소인수분해

/* 소인수분해

 - 소인수분해란 어떤 수를 소수들의 곱으로 표현하는 것입니다. 예를 들어 12를 소인수 분해하면 2 * 2 * 3 으로 나타낼 수 있습니다. 따라서 12의 소인수는 2와 3입니다. 자연수 n이 매개변수로 주어질 때 n의 소인수를 오름차순으로 담은 배열을 return하도록 solution 함수를 완성해주세요.
*/
func solution18(_ n:Int) -> [Int] {
    if n == 2 { return [2] }
    var result:[Int] = []
    var num = n
    for i in 2...n {
        if num%i == 0 {
            result.append(i)
            while num%i == 0 {
                num /= i
            }
        }
        if num == 1 {
            break
        }
    }
    return result
}