[프로그래머스-Lv1] 키패드 누르기 / Python

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

import sys

input = sys.stdin.readline
numbers = list(map(int, input().split()))
hand = input()


def solution(numbers, hand):
    lastL, lastR = 10, 12
    answer = ""
    for n in numbers:
        if n in [1, 4, 7]:
            lastL = n
            answer += "L"
        elif n in [3, 6, 9]:
            lastR = n
            answer += "R"
        else:
            n = 11 if n == 0 else n
            absL = abs(n - lastL)
            absR = abs(n - lastR)
            if sum(divmod(absL, 3)) > sum(divmod(absR, 3)):
                lastR = n
                answer += "R"
            elif sum(divmod(absL, 3)) < sum(divmod(absR, 3)):
                lastL = n
                answer += "L"
            else:
                if hand == "left":
                    lastL = n
                    answer += "L"
                else:
                    lastR = n
                    answer += "R"
    return answer


print(solution(numbers, hand))

 

해설

이 문제에서 생각해야하는 부분은 키패드의 가운데 라인이다.

1) 눌러야하는 숫자와 왼손 엄지손가락에 위치한 부분

2) 눌러야하는 숫자와 오른손 엄지손가락에 위치한 부분

 

두 숫자 차이의 절댓값을 3으로 나눈 몫과 나머지의 합이 원래 위치에서 눌러야하는 숫자까지 이동해야하는 칸이다.