[Programmers] Lv2. 메뉴 리뉴얼(kotlin)

문제

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

 

 

 

예시

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

 

 

풀이

• 하나의 주문(order)에서 구성될 수 있는 메뉴 구성을 백트래킹 기법을 사용합니다. 이 때 최소 2개 이상의 메뉴가 구성되어야 하기에 문자열이 2 이하인 경우는 제외하고 오름차순으로 정렬로 표현해야하기에 미리 정렬을 시킵니다.
• 이후 course 를 하나씩 탐색하면서 해당 갯수로 구성된 메뉴 조합 중 2번 이상 주문된  것 중 최댓값을 구합니다. 그리고 최댓값과 동일한 메뉴 구성을 모두 answer에 추가하시면 됩니다.

class Solution {
    val csmap = mutableMapOf<String, Int>()
    
    fun getCourseInfo(str: String, idx: Int, s: String){
        if(str.length == idx){
            if(s.length >= 2){
                csmap.put(s, csmap.getOrDefault(s, 0) + 1)
            }
        }else{
            getCourseInfo(str, idx+1, s+str[idx])
            getCourseInfo(str, idx+1, s)
        }
    }
    
    fun solution(orders: Array<String>, course: IntArray): Array<String> {  
        val answer = mutableListOf<String>()
        orders.forEach{
            val str = it.toCharArray().sorted().joinToString("")
            getCourseInfo(str, 0, "")
        }
        
        course.forEach{ cnt ->
            val max = csmap.filter{it.key.length == cnt && it.value >= 2}.values.maxOrNull()
            max?.let{ max ->
                answer.addAll(csmap.filter{it.key.length == cnt}.filter{it.value == max}.keys)
            }
        }

        return answer.sorted().toTypedArray()
    }
}

 

 

 

참고

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

maxOrNull - Kotlin Programming Language