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

프로그래머스

---

나의 풀이

import Foundation

// 소수판별 함수
func isPrimeNumber(x: Int) -> Bool {
    if x < 4 { return x == 1 ? false : true }
    for i in 2...Int(sqrt(Double(x))) {
        if (x % i == 0) { return false }
    }
    return true
}

func solution(_ n:Int, _ k:Int) -> Int {
    let arr = String(n,radix: k).split(separator: "0").map { Int($0)! }
    var result = 0
    for i in arr {
        if isPrimeNumber(x: i) { result += 1 }
    }
    return result
}

풀이 키워드

소수판별하기
x라는 Int값의 소수를 구하고 싶다면
for i in 2...(x의 제곱근)까지 반복하며
x % i 가 0이 나온다면 소수가 될 수 없고, 단 한번도 0이 안나온다면 소수로 판별할 수 있다.

 

---