(Kotlin) 프로그래머스 - 키패드 누르기 [2020 카카오 인턴십]

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

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

 

문제 설명

스마트폰 전화 키패드의 각 칸에 다음과 같이 숫자들이 적혀 있습니다.

이 전화 키패드에서 왼손과 오른손의 엄지손가락만을 이용해서 숫자만을 입력하려고 합니다.
맨 처음 왼손 엄지손가락은 *키패드에 오른손 엄지손가락은 #키패드 위치에서 시작하며, 엄지손가락을 사용하는 규칙은 다음과 같습니다.

1. 엄지손가락은 상하좌우 4가지 방향으로만 이동할 수 있으며 키패드 이동 한 칸은 거리로 1에 해당합니다.
2. 왼쪽 열의 3개의 숫자1,4,7을 입력할 때는 왼손 엄지손가락을 사용합니다.
3. 오른쪽 열의 3개의 숫자3,6,9를 입력할 때는 오른손 엄지손가락을 사용합니다.
4. 가운데 열의 4개의 숫자2,5,8,0을 입력할 때는 두 엄지손가락의 현재 키패드의 위치에서 더 가까운 엄지손가락을 사용합니다.
   4-1. 만약 두 엄지손가락의 거리가 같다면, 오른손잡이는 오른손 엄지손가락, 왼손잡이는 왼손 엄지손가락을 사용합니다.

순서대로 누를 번호가 담긴 배열 numbers, 왼손잡이인지 오른손잡이인 지를 나타내는 문자열 hand가 매개변수로 주어질 때, 각 번호를 누른 엄지손가락이 왼손인 지 오른손인 지를 나타내는 연속된 문자열 형태로 return 하도록 solution 함수를 완성해주세요.

[제한사항]

• numbers 배열의 크기는 1 이상 1,000 이하입니다.
• numbers 배열 원소의 값은 0 이상 9 이하인 정수입니다.
• hand는"left" 또는"right" 입니다.
  • "left"는 왼손잡이,"right"는 오른손잡이를 의미합니다.
• 왼손 엄지손가락을 사용한 경우는L, 오른손 엄지손가락을 사용한 경우는R을 순서대로 이어붙여 문자열 형태로 return 해주세요.

---

입출력 예

numbers | hand | result

[1, 3, 4, 5, 8, 2, 1, 4, 5, 9, 5]	"right"	"LRLLLRLLRRL"
[7, 0, 8, 2, 8, 3, 1, 5, 7, 6, 2]	"left"	"LRLLRRLLLRR"
[1, 2, 3, 4, 5, 6, 7, 8, 9, 0]	"right"	"LLRLLRLLRL"

입출력 예에 대한 설명

입출력 예 #1

순서대로 눌러야 할 번호가 [1, 3, 4, 5, 8, 2, 1, 4, 5, 9, 5]이고, 오른손잡이입니다.

왼손 위치 | 오른손 위치 | 눌러야 할 숫자 | 사용한 손 | 설명

*	#	1	L	1은 왼손으로 누릅니다.
1	#	3	R	3은 오른손으로 누릅니다.
1	3	4	L	4는 왼손으로 누릅니다.
4	3	5	L	왼손 거리는 1, 오른손 거리는 2이므로 왼손으로 5를 누릅니다.
5	3	8	L	왼손 거리는 1, 오른손 거리는 3이므로 왼손으로 8을 누릅니다.
8	3	2	R	왼손 거리는 2, 오른손 거리는 1이므로 오른손으로 2를 누릅니다.
8	2	1	L	1은 왼손으로 누릅니다.
1	2	4	L	4는 왼손으로 누릅니다.
4	2	5	R	왼손 거리와 오른손 거리가 1로 같으므로, 오른손으로 5를 누릅니다.
4	5	9	R	9는 오른손으로 누릅니다.
4	9	5	L	왼손 거리는 1, 오른손 거리는 2이므로 왼손으로 5를 누릅니다.
5	9	-	-

따라서 "LRLLLRLLRRL"를 return 합니다.

입출력 예 #2

왼손잡이가 [7, 0, 8, 2, 8, 3, 1, 5, 7, 6, 2]를 순서대로 누르면 사용한 손은 "LRLLRRLLLRR"이 됩니다.

입출력 예 #3

오른손잡이가 [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]를 순서대로 누르면 사용한 손은 "LLRLLRLLRL"이 됩니다.

 

풀이

레벨1이라서 어렵진 않았지만, 레벨1인데 코드가 길어지는 것 같아서 살짝 당황스러웠던 문제다.

 

숫자들이 주어졌을 때 각 숫자를 눌러야 하는 손가락의 배열을 반환하면 되는데,

가운데 숫자인 2, 5, 8, 0에 대한 처리가 필요하다.

 

문제에서 주어진 것처럼 거리를 계산하기 위해, 키패드에 대응하는 좌표계를 설정했다.

그리고 가운데 숫자와 각 손가락 좌표의 거리를 계산해야 하므로, 각 손가락의 위치를 나타내는 변수를 추가했다.

거리를 계산할 때는 얼마나 떨어져 있는지만 파악하기 위해 루트와 제곱 계산은 생략하고, x좌표 차이와 y좌표 차이의 합으로만 처리했다.  

 

코드

class Solution {
    data class Pos(val x: Int, val y: Int)  // 좌표 

    // 키패드에 대응하는 좌표 맵
    val coordinateMap = mapOf(
        "1" to Pos(0, 0),
        "2" to Pos(1, 0),
        "3" to Pos(2, 0),
        "4" to Pos(0, 1),
        "5" to Pos(1, 1),
        "6" to Pos(2, 1),
        "7" to Pos(0, 2),
        "8" to Pos(1, 2),
        "9" to Pos(2, 2),
        "*" to Pos(0, 3),
        "0" to Pos(1, 3),
        "#" to Pos(2, 3)
    )
    var leftThumb = coordinateMap["*"]  // 왼손 위치 좌표
    var rightThumb = coordinateMap["#"] // 오른손 위치 좌표
    var hand = "R"                      // 손잡이 임시 초기화


    fun solution(numbers: IntArray, hand: String): String {
        var answer = ""
        this.hand = when(hand){     // (오른/왼)손잡이 초기화
            "left" -> "L"
            else -> "R"
        }

        numbers.forEach{        // 키패드 번호에 맞는 손가락을 찾는다
            when(it){
                // 1, 4, 7 -> 왼손
                1, 4, 7 -> {
                    answer += 'L'
                    leftThumb = coordinateMap[it.toString()]
                }

                // 3, 6, 9 -> 오른손
                3, 6, 9 -> {
                    answer += 'R'
                    rightThumb = coordinateMap[it.toString()]
                }

                // 가운데 키패드
                else ->{
                    answer += getAnswer(it.toString())  // 각 손가락 위치와 거리를 계산해서 정답 추가
                    when(answer.last()){
                        'L' -> leftThumb = coordinateMap[it.toString()]
                        'R' -> rightThumb = coordinateMap[it.toString()]
                    }

                }
            }
        }
        return answer
    }

    // 각 손가락 위치와 키패드와의 거리를 계산해서 정답 반환
    private fun getAnswer(number: String): String{
        val numberPos: Pos = coordinateMap[number]!!    // 키패드 -> 좌표 변환
        
        // 얼마나 떨어져 있는지만 계산하면 되니까 루트랑 제곱 계산은 생략
        val leftDistance =
            (numberPos.x - leftThumb!!.x).absoluteValue + (numberPos.y - leftThumb!!.y).absoluteValue
        val rightDistance =
            (numberPos.x - rightThumb!!.x).absoluteValue + (numberPos.y - rightThumb!!.y).absoluteValue

        return when{
            leftDistance > rightDistance -> "R"
            leftDistance < rightDistance -> "L"
            else -> hand
        }
    }
}

 

리뷰

forEach문을 함수로 따로 빼서 만들면 solution 함수가 짧아져서 읽기에 더 좋을 것 같다.

 

문제 풀 때는 코드가 길어져서 이거 레벨1 맞나? 싶었는데, 다 해결하고 글로 적으려니까 풀이가 정말 직관적이고 문제 설명에 나온 그대로라서 특별히 쓸 게 없어서 머쓱했다ㅋㅋ

레벨1이 맞는 걸로.