[Swift] k진수에서 소수 개수 구하기

import Foundation

func solution(_ n:Int, _ k:Int) -> Int {
    String(n, radix: k).split(separator: "0").map { Int($0)! }.filter { isPrime($0) }.count
}

func isPrime(_ n: Int) -> Bool {
    if n <= 1 {
        return false
    }
    var i = 2
    while i < Int(sqrt(Double(n))) + 1 {
        if n % i == 0 {
            return false
        }
        i += 1
    }
    return true
}