[Programmers] Lv2.k진수에서 소수 개수 구하기(kotlin)

문제

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

 

 

 

예시

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

 

 

 

풀이

• 소수 찾기 알고리즘은 정말 많이 있으니 생략하겠습니다.
• 해당 문제에서는 k진수로 변경 시에 Int의 크기 자리수보다는 커지기 때문에 소수 판별함수에서는 Long으로 선언해야 합니다.
• StringBuilder에 하나씩 나눠서 추가했지만, toString(k) 라는 메소드를 사용하면 바로 구할 수 있다고 합니다. 😅 역시 또 다시 한번 함수를 많이 알아야 몸이 고생하지 않는다는 것을 깨닫게 되네요 .

import kotlin.math.*

class Solution {
    fun isPrime(n: Long): Boolean{
        if(n == 1L) return false
        
        (2..sqrt(n.toDouble()).toLong()).forEach{
            if(n%it == 0L) return false
        }
        return true
    }
    
    
    fun solution(n: Int, k: Int): Int {
        var answer: Int = -1
        
        val str = StringBuilder().apply{
            var num = n
            while(num>0){
                append(num%k)
                num/=k
            }
            reverse()
        }
        
        return str.split("0").filter{it.isNotEmpty()}.filter{isPrime(it.toLong())}.size
    }
}

 

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• 참고로 String에는 reversed() 메소드이고 StringBuilder는 reverse() 입니다~