[카카오 인턴] 키패드 누르기

코틀린

programmers.co.kr/learn/courses/30/lessons/67256

코딩테스트 연습 - 키패드 누르기

class Solution {
    fun solution(numbers: IntArray, hand: String): String {
        var answer = ""

        val map = mutableMapOf(
            1 to Pair(0, 0),
            2 to Pair(0, 1),
            3 to Pair(0, 2),
            4 to Pair(1, 0),
            5 to Pair(1, 1),
            6 to Pair(1, 2),
            7 to Pair(2, 0),
            8 to Pair(2, 1),
            9 to Pair(2, 2),
            "*" to Pair(3, 0),
            0 to Pair(3, 1),
            "#" to Pair(3, 2)
        )

        fun Pair<Int, Int>.distance(pair: Pair<Int, Int>): Int =
            abs(this.first - pair.first) + abs(this.second - pair.second)

        var currentL: Any = "*"
        var currentR: Any = "#"

        for (number in numbers) {
            if (setOf(1, 4, 7).contains(number)) {
                answer += "L"
                currentL = number
            } else if (setOf(3, 6, 9).contains(number)) {
                answer += "R"
                currentR = number
            } else {
                val postPosition = map[number]!!

                val a: Int = postPosition.let { map[currentL]!!.distance(it) }

                val b: Int = postPosition.let { map[currentR]!!.distance(it) }

                if (a < b) {
                    answer += "L"
                    currentL = number
                } else if (a == b) {
                    if (hand == "right") {
                        answer += "R"
                        currentR = number
                    } else {
                        answer += "L"
                        currentL = number
                    }
                } else {
                    answer += "R"
                    currentR = number
                }
            }
        }

        return answer
    }
}