프로그래머스 2020 카카오 인턴 - 키패드 누르기 해설 (파이썬)

문제 설명

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

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

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 해주세요.

 

해설

처음 내가 생각했던 문제 풀이 방식은, 키패드를 하나의 행렬로 보고, 왼손가락과 오른손가락이 가있는 좌표를 저장한 후, 현재 눌러야 하는 키패드의 좌표와 각 손가락의 좌표를 가지고 거리를 구해서 가까운 손가락으로 누를 수 있도록 풀이를 하려고 했다. 그래서 거리를 구하는 방식을 유클리디안 거리를 이용해서 구하려고 했으나, 테스트 케이스 13~20번까지가 맞지 않는 상황이 발생했다. 그래서 왜 그런지 생각해보니까, 문제에 "상하좌우"로만 움직인다는 조건이 포함되어있었다. 따라서 이문제는 손가락의 위치와 현재 눌러야 하는 키패드의 좌표의 x 이동량 + y 이동량을 더해서 비교를 해야한다. 우선 코드는 아래와 같다.

keypad = [[1, 4, 7, '*'], [2, 5, 8, 0], [3, 6, 9, '#']] # (1)
left = [0, 3] # (2)
right = [2, 3] 


def solution(numbers, hand):
    answer = ''

    for n in numbers:
        for idx, v in enumerate(keypad): # (3)
            if n in v:
                co = v.index(n) # (4)
                if idx == 0:
                    answer += 'L'
                    left[0] = 0
                    left[1] = co
                elif idx == 2:
                    answer += 'R'
                    right[0] = 2
                    right[1] = co
                else:
                    now = [1, co]
                    left_distance = abs(now[0] - left[0]) + abs(now[1] - left[1]) # (5)
                    right_distance = abs(now[0] - right[0]) + abs(now[1] - right[1])
                    if left_distance > right_distance:
                        answer += 'R'
                        right[0] = 1
                        right[1] = co
                    elif right_distance > left_distance:
                        answer += 'L'
                        left[0] = 1
                        left[1] = co
                    else:
                        if hand == "right":
                            answer += 'R'
                            right[0] = 1
                            right[1] = co
                        else:
                            answer += 'L'
                            left[0] = 1
                            left[1] = co

    return answer

(1) : keypad라는 이중 배열에 왼손가락으로 누를 수 있는 키패드, [2, 5, 8, 0], 오른손가락으로 누를 수 있는 키패드 배열을 넣어놨다.

(2) : 왼손가락과 오른손가락이 현재 있는 좌표를 저장하기 위한 배열로, 처음에는 각각 '*', '#'에서 시작하므로, *과 #의 인덱스를 넣어줬다. 

(3) : enumerate() 메소드를 활용해서 keypad의 index와 value가 같이 나오도록 하였다.

(4) : index() 메소드를 사용하면 index() 메소드의 파라미터 값이 배열 어디 인덱스에 위치해있는지 알 수 있다.

(5) : 현재 좌표 (now)와 왼손가락 (left)의 각각 x좌표, y좌표의 차를 구한후 abs() 메소드를 이용해서 절대값으로 바꿔준다. (방향 없이 순수하게 이동량만 측정하므로)

 

정리하자면,

1. numbers를 돌면서 keypad 내부의 배열들에 numbers 배열의 요소가 있는지 찾는다.

2. 왼손가락으로 입력할 수 있는 좌표일 경우 왼손가락의 현재 좌표를 저장해주고 answer에는 'L'을 더해준다. 오른손가락 역시 오른손가락의 현재 좌표를 저장해주고 answer에는 'R'을 더해준다. 2, 5, 8, 0 중 하나일 경우에는 왼손가락으로 입력할건지 오른손가락으로 입력할건지 정해야한다.

3. (5)번 로직을 통해서 x의 이동량, y의 이동량의 총합을 구해서 어디가 더 가까운지 판별해서 가까운 손가락으로 입력하도록 한다. 

4. 왼손가락, 오른손가락의 거리가 같을 경우 왼손잡이인지 오른손잡이인지 판별해서 좌표를 저장하고, 문자를 더해준다. 왼손잡이일경우 'L', 오른손잡이일경우 'R'

 

난이도가 크게 어려운 문제는 아니었으나, if, elif, else로 떡칠한 코드를 짠거 같아서 더 깔끔하게 짤수는 없을지 고민해봐야겠다.