[ 프로그래머스 - Kotlin ] 소수 찾기 ( 코틀린 )

( 완전탐색 / 소수 찾기 )

[문제]

문제 설명

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

제한사항

• numbers는 길이 1 이상 7 이하인 문자열입니다.
• numbers는 0~9까지 숫자만으로 이루어져 있습니다.
• "013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

입출력 예시

numbers	return
"17"	3
"011"	2

예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.

예제 #2
[0, 1, 1]으로는 소수 [11, 101]를 만들 수 있습니다.

• 11과 011은 같은 숫자로 취급합니다.

---

[코드]

class Solution {
    fun solution(numbers: String): Int {
        var answer = 0
        val numSet = HashSet<Int>()
        // 모든 경우의 숫자 set을 생성
        makeNumberSet("", numbers, numSet)
        // HashSet의 모든 항목을 반복
        while(numSet.iterator().hasNext()){
            // numSet항목을 꺼낸 후 제거
            val num = numSet.iterator().next()
            numSet.remove(num)
            // 해당 숫자가 소수면 answer 증가
            if(isPrime(num)) answer++
        }
        
        return answer
    }
    
    fun makeNumberSet(prefix:String, numbers:String, numSet:HashSet<Int>){
        // 저장할 값이 있을 경우 numSet에 추가
        if(prefix != "") numSet.add(prefix.toInt())
        for(i in numbers.indices){
            // 현재 값에서 다음 값을 추가 후 numbers에서 추가한 값을 제외시킴
            makeNumberSet("$prefix${numbers[i]}", numbers.substring(0, i) + numbers.substring(i+1), numSet)
        }
    }
    
    fun isPrime(num:Int):Boolean{
        // 1이하는 소수가 아님으로 false반환
        if(num <= 1) return false
        // 소수를 판별할 숫자의 제곱근 까지 나누어서 나누어 떨어지지 않으면 해당 숫자는 소수임으로 none으로 확인 후 결과 리턴
        return (2..Math.sqrt(num.toDouble()).toInt()).none{ num % it == 0 }
    }
}

문제 링크

코딩테스트 연습 - 소수 찾기

제가 잘못 알고 있거나 잘못된 부분이 있을 경우 알려주시고 추가로 궁금한 점 있으신 분들도 댓글이나 메일 주시면 성실히 답변해 드리겠습니다.🧑🏻‍💻

감사합니다~😄