[프로그래머스 • 코틀린] Generating Prime Number #12977

#12977

🎄 Question ?

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

class Solution {
    fun solution(nums: IntArray): Int {
        var answer = -1
 
        // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
        println("Hello Kotlin")

        return answer
    }
}

🧩 Thought Process

1. nums 배열에서 3개의 값을 더할 수 있는 모든 경우의 수를  for 문을 이용하여 찾아주었습니다.. 
2. nums 배열에서 3개의 값을 더했을 때 생기는 소수의 개수를 세는 문제인줄 알았는데 nums 배열에서 3개의 값을 더했을 때 소수가 생기는 경우의 수를 물어보는 문제였습니다. 문제를 정말 꼼꼼히 읽어야 할 것 같습니다.
3. 그래서 아래 코드에는 중복되는 소수를 제거하기 위해 mutableSet을 사용했었는데 mutableList로 바꾸어 중복되는 경우도 세주니까 코드가 성공적으로 실행 되었습니다.
4. forEach문을 사용하여 소수의 개수를 셈해 주었습니다.
5. isPrimeNumber이라는 함수를 만들어 소수를 판별해주었습니다.

 

class Solution {
    fun solution(nums: IntArray): Int {
        var answer = 0
        var set : MutableSet<Int> = mutableSetOf()
 
        // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
        println("Hello Kotlin")
        for (i in 0 until nums.size - 2 ) {
            for (j in (i+1) until nums.size - 1) {
                for (k in (j+1) until nums.size) {
                    set.add(nums[i] + nums[j] + nums[k])
                }
            }
        }
        println(set)
        
        set.forEach {
            when (isPrimeNumber(it)) {
                true -> answer += 1
            }
        }
        return answer
    }
    
    fun isPrimeNumber(num: Int): Boolean {
        for (i in 2 until num) {
            when {
                (num%i == 0) -> return false
            }
        }
        return true
    }
}

🎀 Answer

class Solution {
    fun solution(nums: IntArray): Int {
        var answer = 0
        var set : MutableList<Int> = mutableListOf()
 
        // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
        println("Hello Kotlin")
        for (i in 0 until nums.size - 2 ) {
            for (j in (i+1) until nums.size - 1) {
                for (k in (j+1) until nums.size) {
                    set.add(nums[i] + nums[j] + nums[k])
                }
            }
        }
        println(set)
        
        set.forEach {
            when (isPrimeNumber(it)) {
                true -> answer += 1
            }
        }
        return answer
    }
    
    fun isPrimeNumber(num: Int): Boolean {
        for (i in 2 until num) {
            when {
                (num%i == 0) -> return false
            }
        }
        return true
    }
}

 

🎁 Result

🏆 Comment