[프로그래머스/Swift] k진수에서 소수 개수 구하기

 

https://school.programmers.co.kr/learn/courses/30/lessons/92335

프로그래머스

 

String 이니셜라이저를 통해 k진수로 바꿀 수 있다!

조건을 보니 0을 기준으로 문자들을 나누면 될 것 같았다.

그다음 소수인지 판별만 해주면 된다.

 

import Foundation

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

func solution(_ n:Int, _ k:Int) -> Int {
    var answer = 0
    let array = String(n, radix: k).split(separator: "0").map{ String($0) }
    for i in 0..<array.count {
        if isPrime(num: Int(array[i])!) {
            answer += 1
        }
    }
    return answer
}

 

처음엔 에라토스테네스의 체를 사용해서 했는데 core dumped가 일어났다. 범위가 1000000 여서 그런 것 같다.

그래서 서둘러 sqrt를 사용해서 소수 판별을 해줬다.