[프로그래머스] 조이스틱 (42860)(Kotlin)

원본 문제

문제 설명

조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 A로만 이루어져 있습니다. ex) 완성해야 하는 이름이 세 글자면 AAA, 네 글자면 AAAA 조이스틱을 각 방향으로 움직이면 아래와 같습니다.

▲ - 다음 알파벳
▼ - 이전 알파벳 (A에서 아래쪽으로 이동하면 Z로)
◀ - 커서를 왼쪽으로 이동 (첫 번째 위치에서 왼쪽으로 이동하면 마지막 문자에 커서)
▶ - 커서를 오른쪽으로 이동

예를 들어 아래의 방법으로 JAZ를 만들 수 있습니다.

- 첫 번째 위치에서 조이스틱을 위로 9번 조작하여 J를 완성합니다.
- 조이스틱을 왼쪽으로 1번 조작하여 커서를 마지막 문자 위치로 이동시킵니다.
- 마지막 위치에서 조이스틱을 아래로 1번 조작하여 Z를 완성합니다.
따라서 11번 이동시켜 “JAZ”를 만들 수 있고, 이때가 최소 이동입니다.

만들고자 하는 이름 name이 매개변수로 주어질 때, 이름에 대해 조이스틱 조작 횟수의 최솟값을 return 하도록 solution 함수를 만드세요.

제한 사항

• name은 알파벳 대문자로만 이루어져 있습니다.
• name의 길이는 1 이상 20 이하입니다.

예제 입출력

name	return
“JEROEN”	56
“JAN”	23

문제 풀이1

import kotlin.math.abs

class Solution {
    var check: BooleanArray = booleanArrayOf()
    var minList: MutableList<Int> = mutableListOf()

    fun solution(name: String): Int {
        minList = MutableList(name.length) { 0 }
        check = BooleanArray(name.length)

        name.toCharArray()
            .forEachIndexed { index, char ->
            minList[index] = Math.min('Z' - char + 1, char - 'A')
            if (minList[index] == 0) check[index] = true
        }

        var next = 0
        var result = 0
        while (check.count { it } != name.length) {
            val (nextIndex: Int, nextDistance: Int) = dijkstra(next, name.length)
            next = nextIndex
            result += nextDistance + minList[nextIndex]

        }

        return result
    }

    fun dijkstra(start: Int, nameLength: Int): IntArray {
        check[start] = true

        var distance = IntArray(nameLength) { 0 }
        var minIndex = 0
        var minCount = Int.MAX_VALUE

        for (i in 0 until nameLength) {
            if (i == start || check[i]) continue
            else {
                distance[i] = Math.min(abs(i - start), start + nameLength - i)
                if (minCount > distance[i]) {
                    minCount = distance[i]
                    minIndex = i
                }
            }
        }

        return if (minCount == Int.MAX_VALUE) intArrayOf(0, 0)
        else intArrayOf(minIndex, minCount)
    }
}