[Programmers] Lv1. 대충 만든 자판(kotlin)

문제

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

 

 

 

예시

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

 

 

풀이

• targets 원소 하나당 최소 키를 몇 번 누르는지 알아야 합니다.
• 문자열의 문자 하나씩 keymap에서 조회하여 가장 index가 작은 것을 택해야 합니다.
  하지만, 이 때 모든 keymap에 찾고자 하는 문자가 없을 경우는 문자열 자체를 keymap을 통해서 만들 수 없다는 말과 같습니다.
• 따라서, check 변수를 통해 하나의 문자에서 찾을 수 없을 경우는 -1을 반환하도록 해야합니다.
• 문자열을 찾을 수 있을 때 fold 연산자를 통해서 (index + 1)을 더한 값을 반환하면 됩니다.

import kotlin.math.*

class Solution {
    fun solution(keymap: Array<String>, targets: Array<String>): IntArray = targets.map{ target ->
            var check = true
            val sum = target.fold(0){ acc, ch -> 
                val minIndex = keymap.minByOrNull{ 
                    if(it.indexOf(ch) == -1) Int.MAX_VALUE 
                    else it.indexOf(ch) 
                }?.indexOf(ch) ?: -1
                
                if(minIndex == -1) check = false
                acc + minIndex + 1
            }
            if(check) sum else -1
        }.toIntArray()
}

 

 

참고

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