[프로그래머스] (Swift) kakao - k진수에서 소수 개수 구하기 (lv.2)

🟠 문제 링크

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

프로그래머스

 

🟠 문제 풀이

1. k진수로 바꾸면서 동시에 배열로 만들어주는데, 0으로 split 해서 넣어주기
2. split해서 넣어준 배열에서 하나씩 꺼내서 소수 판별
  - 0은 split 단어이기 때문에 pass
  - 1은 소수가 아니기 때문에 pass
  - 소수 판별해줘서 소수라면 true, 아니라면 false
3. 소수이면 count += 1  해줌

 

🟠 소스코드

https://github.com/deslog/Algorithm/blob/main/Algorithm/Programmers/kakao_k%EC%A7%84%EC%88%98%EC%97%90%EC%84%9C%20%EC%86%8C%EC%88%98%20%EA%B0%9C%EC%88%98%20%EA%B5%AC%ED%95%98%EA%B8%B0/main.swift

GitHub - deslog/Algorithm

import Foundation

func solution(_ n: Int, _ k: Int) -> Int {
    let changeN = changeFormat(n: n, k: k)
    var count = 0

    for char in changeN {
        if char == 0 {
            continue
        }
        if char == 1 {
            continue
        }
        if isPrime(n: char) {
            count += 1
        }
    }
    return count
}

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

func changeFormat(n: Int, k: Int) -> [Int] {
    let split = String(n, radix: k).split(separator: "0").map { Int($0)! }
    return split
}