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

린다·2022년 3월 16일
0

Algorithm Practice

목록 보기
8/8
post-thumbnail

🔗 문제 링크

✔️ 문제 설명

양의 정수 n이 주어집니다. 이 숫자를 k진수로 바꿨을 때, 변환된 수 안에 아래 조건에 맞는 소수(Prime number)가 몇 개인지 알아보려 합니다.
0P0처럼 소수 양쪽에 0이 있는 경우
P0처럼 소수 오른쪽에만 0이 있고 왼쪽에는 아무것도 없는 경우
0P처럼 소수 왼쪽에만 0이 있고 오른쪽에는 아무것도 없는 경우
P처럼 소수 양쪽에 아무것도 없는 경우
단, P는 각 자릿수에 0을 포함하지 않는 소수입니다.
예를 들어, 101은 P가 될 수 없습니다.
예를 들어, 437674을 3진수로 바꾸면 211020101011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 왼쪽부터 순서대로 211, 2, 11이 있으며, 총 3개입니다. (211, 2, 11을 k진법으로 보았을 때가 아닌, 10진법으로 보았을 때 소수여야 한다는 점에 주의합니다.) 211은 P0 형태에서 찾을 수 있으며, 2는 0P0에서, 11은 0P에서 찾을 수 있습니다.
정수 n과 k가 매개변수로 주어집니다. n을 k진수로 바꿨을 때, 변환된 수 안에서 찾을 수 있는 위 조건에 맞는 소수의 개수를 return 하도록 solution 함수를 완성해 주세요.

📚 코드

import Foundation

func solution(_ n:Int, _ k:Int) -> Int {
    var answer: Int = 0
    
    let kNotationNum = transInToKnotation(n, k)
    let numList = kNotationNum.components(separatedBy: "0").filter { $0 != "" }
    
    numList.forEach {
        if isPrimeNum($0) {
            answer += 1
        }
    }
    return answer
}

func transInToKnotation(_ n: Int, _ k: Int) -> String {
    var totalResult = ""
    
    var (quot, remainder) = n.quotientAndRemainder(dividingBy: k)
    
    while quot > 0 || remainder > 0 {
        totalResult = "\(remainder)" + totalResult
        (quot, remainder) = quot.quotientAndRemainder(dividingBy: k)
    }
    
    return totalResult
}

func isPrimeNum(_ n: String) -> Bool {    
    let num = Int(n)!
    
    var isPrime = true
    
    if num < 3 {
        isPrime = num == 1 ? false : true
    } else {
        for division in 2...Int(sqrt(Double(num))) + 1 {
            if num % division == 0 {
                isPrime = false
                break
            }
        } 
    }
        
    return isPrime
}

💡 풀이

  • 함수는 크게 solution 함수, k진수로 변환하는 함수, 소수 판별 함수로 구성하였다.
  • 몫과 나머지를 구할 수 있는 quotientAndRemainder(dividingBy:) 메서드를 활용하여 주어진 n을 k진수로 변환한 문자열을 리턴하는 함수를 구성했다.
  • 소수인지 판별하는 함수에 있어서는 1,2만 예외처리를 해주고 나머지 수들은 2부터 자기 자신의 제곱근까지 loop를 돌면서 나머지 0인 경우에는 소수가 아니기 때문에 isPrime을 false로 변환해주었다.

0개의 댓글