프로그래머스 - 키패드 누르기

문제

 

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

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

 

 

풀이

 

[1, 4, 7]을 눌렀을 경우에는 바로 L을 추가하고, [3, 6, 9]를 눌렀을 때는 바로 R을 추가한다.

[2, 5, 8, 0]을 눌렀을 경우에는 왼쪽 손과 오른쪽 손에 있는 버튼과의 거리를 각각 구해서 비교한다. 거리를 구할 때 세로 거리와 가로 거리를 따로 구해서 더해주면 된다. 

 

 

코드

 

python

def solution(numbers, hand):
    answer = ''
    left, right = '*', '#'
    for i in numbers:
        if i in [1, 4, 7]:
            answer += 'L'
            left = i
        elif i in [3, 6, 9]:
            answer += 'R'
            right = i
        else:
            pad = {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]}
            ld = abs(pad.get(i)[0] - pad.get(left)[0]) + abs(pad.get(i)[1] - pad.get(left)[1])
            rd = abs(pad.get(i)[0] - pad.get(right)[0]) + abs(pad.get(i)[1] - pad.get(right)[1])
            if ld < rd:
                left = i
                answer += 'L'
            elif ld > rd:
                right = i
                answer += 'R'
            else:
                answer += hand[0].upper()
                if hand == 'left':
                    left = i
                else:
                    right = i
 
    return answer