[프로그래머스] level 1 키패드 누르기 (파이썬)

• 문제 링크

문제 설명

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

• 엄지손가락은 상하좌우 4가지 방향으로만 이동할 수 있으며 키패드 이동 한 칸은 거리로 1에 해당합니다.
• 왼쪽 열의 3개의 숫자 1, 4, 7을 입력할 때는 왼손 엄지손가락을 사용합니다.
• 오른쪽 열의 3개의 숫자 3, 6, 9를 입력할 때는 오른손 엄지손가락을 사용합니다.
• 가운데 열의 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 해주세요.

문제 풀이

def solution(numbers, hand):
    
    grid = {
        1 : [0,0], 2 : [0,1], 3 : [0,2],
        4 : [1,0], 5 : [1,1], 6 : [1,2],
        7 : [2,0], 8 : [2,1], 9 : [2,2],
        "*": [3,0], 0 : [3,1], "#" :[3,2]
    }
    
    answer = ''
    l,r = "*", "#" 
    for n in numbers:
        if n in [1,4,7]:
            answer += "L"
            l = n
        elif n in [3,6,9]:
            answer += "R"
            r = n 
        # middle number
        else: 
            # calculate the distance
            # |x2-x1| + |y2-y1|
            leftDistance = abs(grid[l][0] - grid[n][0]) + abs(grid[l][1] - grid[n][1])
            rightDistance = abs(grid[r][0] - grid[n][0]) + abs(grid[r][1] - grid[n][1])
            
            if leftDistance > rightDistance:
                answer += "R"
                r = n
            elif rightDistance > leftDistance:
                answer += "L"
                l = n
            else:
                if hand == "right":
                    answer += "R"
                    r = n
                else:
                    answer += "L"
                    l = n                
    
    
    return answer

 

간단한 설명

• 아주 크게 보면:
  • 숫자가 [1,4,7] 안에 있으면 왼손을 움직임
  • 숫자가 [3,6,9] 안에 있으면 오른손을 움직임
  • 숫자가 [2,5,8,0] 안에 있으면 왼손, 오른손 위치를 비교해서 manhatten distance를 통해 leftDistance, rightDistance를 구함
    • leftDistance가 더 클 경우 오른손이 더 가깝단 의미로 오른손을 움직임
    • rightDistance가 더 클 경우 왼손이 더 가까으므로 왼손을 움직임
    • leftDistance = rightDistance일 경우 hand parameter에 의해서 결정됨 
• 내가 생각한 핵심은:
  • 키패드를 map을 사용해서 그리드로 나눈다음에, key는 숫자, value는 숫자의 위치를 넣는다 (list를 사용했지만 tuple도 가능하다)
  • Manhatten Distance를 구할때, 이 키패드 map을 이용해서 각 거리를 계산한다
    • 이때 마지막으로 담은 left, right가 무슨 숫자인지 알아야 하므로 l,r 변수에 계속 업데이트 해준다
  • l,r는 각 "*"과 "#"으로 초기화 시켜주는거 잊지 말자