프로그래머스 | 키패드 누르기 (Python)

프로그래머스

 

내 코드

def solution(numbers, hand):
    answer = ''
    # 키패드 숫자들의 좌표를 정의
    keypad = {
        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)
    }
    
    curr_left = keypad['*']
    curr_right = keypad['#']
    
    for number in numbers:
        if number in [1, 4, 7]:
            curr_left = keypad[number]
            answer += 'L'
        elif number in [3, 6, 9]:
            curr_right = keypad[number]
            answer += 'R'
        else:
            curr = keypad[number]
            
            left_dist = abs(curr[0] - curr_left[0]) + abs(curr[1] - curr_left[1])
            right_dist = abs(curr[0] - curr_right[0]) + abs(curr[1] - curr_right[1])
            
            if right_dist > left_dist:
                curr_left = keypad[number]
                answer += 'L'
            elif right_dist < left_dist:
                curr_right = keypad[number]
                answer += 'R' 
            elif right_dist == left_dist:
                if hand == 'left':
                    curr_left = keypad[number]
                    answer += 'L'
                elif hand == 'right':
                    curr_right = keypad[number]
                    answer += 'R' 
    return answer