[Programmers] Lv2. 피로도(kotlin)

문제

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

 

 

예시

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

 

 

 

풀이

• 백트래킹 기법을 이용하면 손쉽게 풀 수 있습니다.

import kotlin.math.*

class Solution {
    var answer = -1
    
    fun getResult(k: Int, dungeons: Array<IntArray>, check: List<Boolean>){
        var flag = false
        for(i in 0 until check.size){
            if(!check[i] && k >= dungeons[i][0]){
                flag = true
                getResult(k-dungeons[i][1], dungeons, check.toMutableList().apply{ this[i] = true })
            }
        }
        
        if(!flag){
            answer = max(answer, check.count{it})
        }
    }
    
    fun solution(k: Int, dungeons: Array<IntArray>): Int {
        getResult(k, dungeons, List(dungeons.size){false})
        return answer
    }
}

 

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽