[Programmers] Lv1.소수 만들기(kotlin)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12977?language=kotlin#

 

예시

https://school.programmers.co.kr/learn/courses/30/lessons/12977?language=kotlin#

 

풀이

• 소수 찾는 알고리즘은 다양하게 존재하지만 O(n2)의 시간복잡도가 소요되는 알고리즘은 지양해야 합니다. ex) 에라토스테네스의 체
• 해당 문제를 백트래킹을 사용해도 되지만 서로 다른 3개의 숫자가 고정적이기 때문에 3중 for문으로 푸는게 더 간단합니다.

import kotlin.math.*

class Solution {
    
    fun isPrime(n: Int): Boolean = when(n){
        1 -> false
        else -> (2..sqrt(n.toDouble()).toInt()).all{
            n % it != 0
        }
    }
    
    fun solution(nums: IntArray): Int {
        var answer = 0
 
        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){
                    if(isPrime(nums[i] + nums[j] + nums[k])){
                        answer++
                    }
                }
            }
        }
        

        return answer
    }
}

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.sequences/all.html

all - Kotlin Programming Language