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

프로그래머스

문제 설명

양의 정수 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 함수를 완성해 주세요.

제한 조건

• 1 ≤ n ≤ 1,000,000
• 3 ≤ k ≤ 10

​입출력 예

n	k	result
437674	3	3
110011	10	2

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

110011을 10진수로 바꾸면 110011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 11, 11 2개입니다. 이와 같이, 중복되는 소수를 발견하더라도 모두 따로 세어야 합니다.

 

---

 

나의 풀이

1. 입력받은 n을 k진수로 변환하고 0을 기준으로 split한다.

2. split된 배열울 빈값이 아닌 값들만 filter하고, map을 이용해 long으로 변환한다.

　(k진수로 변환한 값이 int 범위를 벗어날 수 있기 때문)

3. transfer의 값들이 소수인지 판별하는 for문을 작성한다.

4. 범위가 커지기 때문에 '제곱근' 까지만 소수인지 판별하면 된다.

import kotlin.math.sqrt

class Solution {
    fun solution(n: Int, k: Int): Int {
        var answer = 0
        val transfer = n.toString(k).split("0").filter { it != "" }.map {it.toLong()}

        for (i in transfer.indices) {
            if (transfer[i] != 1.toLong()) {
                if (isDecimalCalc(transfer[i])) answer++
            }
        }

        return answer
    }

    private fun isDecimalCalc(n: Long): Boolean {
        var isDecimal = true
        for (i in 2..sqrt(n.toDouble()).toInt()) {
            if (n % i == 0.toLong())  {
                isDecimal = false
                break
            }
        }
        return isDecimal
    }
}

처음엔 소수인지 여부를 '절반' 값을 기준으로 for문을 돌렸더니 런타임 에러가 났다.

알고보니 숫자가 너무 커서 '절반'이 아닌 '제곱근'을 기준으로 체크해야 한다고 해서 수정하였다.

또한 k진수로 변환한 값이 int의 범위보다 커져서 런타임 오류가 날 수 있으므로 long으로 필히 변환해야한다.

질문하기가 없었으면 놓칠뻔 한 조건이었다...ㅜ