[프로그래머스 • 코틀린] 기사단원의 무기 #136798

#136798

🎄 Question ?

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

🧩 Thought Process

1. 소수의 개수를 셈할 수 있는 함수를 따로 만들고
2. 소수의 개수가 제한 수 보다 크면 공격 수를 무게로 더해준다.
3. 소수의 개수를 찾을 때 시간 복잡도를 고려해 주어야 한다.

class Solution {
    fun solution(number: Int, limit: Int, power: Int): Int {
        var answer: Int = 0
        for (i in 1..number) {
            var nLCD = countLCD(i)
             when  {
                (nLCD > limit) -> answer += power
                else -> answer += nLCD
            }
        }
        
        return answer
    }
    
    fun countLCD(number: Int): Int {
        var count = 0
        for (i in 1..number) {
            when {
                (number % i == 0) -> count += 1
            }
         
        }
        println(count)
        return count  
    }
}

🎀 Answer

import kotlin.math.sqrt

class Solution {
    fun solution(number: Int, limit: Int, power: Int): Int {
        var answer: Int = 0
        for (i in 1..number) {
            var nLCD = countLCD(i)
             when  {
                (nLCD > limit) -> answer += power
                else -> answer += nLCD
            }
        }
        
        return answer 
    }
    
    fun countLCD(number: Int): Int {
        var count = 0
        if (number == 1) {
            count += 1
        } else {
            count += 2
            for (i in 2 .. sqrt(number.toDouble()).toInt()) {
                if (number%i == 0)  {
                    when {
                        (i*i == number) -> count += 1
                        else -> count += 2
                    }
                }
            }
        }
        return count 
    }
}

 

🎁 Result

🏆 Comment