소수 찾기(Lv. 2)

문제 링크

풀이

import Foundation

func solution(_ numbers:String) -> Int {
    let numArr = numbers.map { Int(String($0))! } // [Int] 형으로 변환
    var resultArr = Set<Int>()
    Recursive(numArr, [], &resultArr)
    
    return resultArr.count
}

func Recursive(_ numArr: [Int], _ numbers: [Int], _ resultArr: inout Set<Int>) {
    for (i, num) in numArr.enumerated() {
        var newNumArr = numbers
        newNumArr.append(num)
        
        var assembleNum = 0
        for (j, num) in newNumArr.enumerated()  {
            assembleNum += Int(pow(Double(10), Double(j))) * num
        }
        
        // 소수 판별
        if isPrime(assembleNum) {
            resultArr.update(with: assembleNum)
        }
        
        var arr = numArr
        arr.remove(at: i)
        Recursive(arr, newNumArr, &resultArr)
    }
}

// 소수 체크
func isPrime(_ num: Int) -> Bool {
    if num == 0 {
        return false
    } else if num < 4 {
        return num == 1 ? false : true
    }
    
    for i in 2...Int(sqrt(Double(num))) {
        if num % i == 0 { 
            return false 
        }
    }
    
    return true
}

후기

완전탐색 후 가능한 소수 값들을 전부 찾는 문제.
재귀로 조합 가능한 경우의 수를 전부 찾았고 Set으로 겹치는 값들을 뺀 나머지 결과값들을 도출했다.
딱히 복잡할 건 없었던 문제.