[Programmers] Lv2. 타겟 넘버(kotlin)

문제

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

 

 

예시

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

 

 

풀이

• DFS 기본적인 문제 유형으로 재귀의 break point만 잘 지정하면 됩니다.

class Solution {
    var answer = 0
    
    fun getresult(numbers: IntArray, checks: IntArray, target: Int){
        if(checks.size == numbers.size){
            val sum = numbers.foldIndexed(0){index, acc, i ->
                acc + i*checks[index]
            }
            
            if(target == sum) answer++
        }else{
            getresult(numbers, checks + intArrayOf(-1), target)
            getresult(numbers, checks + intArrayOf(1), target)
        }
    }
    
    fun solution(numbers: IntArray, target: Int): Int {
        getresult(numbers, intArrayOf(), target)
        return answer
    }
}

 

 

 

참고

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

foldIndexed - Kotlin Programming Language