[ Programmers / CodingTest / Python ] 키패드 누르기

황승환·2022년 1월 18일
0

Python

목록 보기
104/498

문제 설명

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

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

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

입출력 예

numbers					hand		result
[1, 3, 4, 5, 8, 2, 1, 4, 5, 9, 5]	"right"		"LRLLLRLLRRL"
[7, 0, 8, 2, 8, 3, 1, 5, 7, 6, 2]	"left"		"LRLLRRLLLRR"
[1, 2, 3, 4, 5, 6, 7, 8, 9, 0]		"right"		"LLRLLRLLRL"

입출력 예에 대한 설명

입출력 예 #1

순서대로 눌러야 할 번호가 [1, 3, 4, 5, 8, 2, 1, 4, 5, 9, 5]이고, 오른손잡이입니다.

왼손 위치	오른손 위치	눌러야 할 숫자	사용한 손	설명
*	#	1	L		1은 왼손으로 누릅니다.
1	#	3	R		3은 오른손으로 누릅니다.
1	3	4	L		4는 왼손으로 누릅니다.
4	3	5	L		왼손 거리는 1, 오른손 거리는 2이므로 왼손으로 5를 누릅니다.
5	3	8	L		왼손 거리는 1, 오른손 거리는 3이므로 왼손으로 8을 누릅니다.
8	3	2	R		왼손 거리는 2, 오른손 거리는 1이므로 오른손으로 2를 누릅니다.
8	2	1	L		1은 왼손으로 누릅니다.
1	2	4	L		4는 왼손으로 누릅니다.
4	2	5	R		왼손 거리와 오른손 거리가 1로 같으므로, 오른손으로 5를 누릅니다.
4	5	9	R		9는 오른손으로 누릅니다.
4	9	5	L		왼손 거리는 1, 오른손 거리는 2이므로 왼손으로 5를 누릅니다.
5	9	-	-	

따라서 "LRLLLRLLRRL"를 return 합니다.

입출력 예 #2

왼손잡이가 [7, 0, 8, 2, 8, 3, 1, 5, 7, 6, 2]를 순서대로 누르면 사용한 손은 "LRLLRRLLLRR"이 됩니다.

입출력 예 #3

오른손잡이가 [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]를 순서대로 누르면 사용한 손은 "LLRLLRLLRL"이 됩니다.

접근 방법

우선 손가락의 위치를 각각 저장할 변수 left, right를 선언하고 *와 #를 표현하기 위해 문자열로 numbers의 각 자릿수들을 처리하도록 했다. numbers를 deque로 변환하여 popleft()를 사용하여 성능을 높였다. 왼손의 범위의 수나 오른손의 범위의 수가 나오면 left, right를 갱신시켜주고 만약 가운데 수 (2, 5, 8, 0)이 나오면 좌표간의 거리를 계산하는 함수를 작성하고 사용하여 어떤 손가락으로 누를지를 결정하도록 하였다.

  • 정답을 저장할 문자열 answer를 선언한다.
  • leftpop()을 사용하기 위해 numbers를 deque로 변환한다.
  • 왼손의 범위를 저장할 배열 l에 1, 4, 7을 문자열 형태로 넣어준다.
  • 오른손의 범위를 저장할 배열 r에 3, 6, 9를 문자열 형태로 넣어준다.
  • 왼손과 오른손의 현재 위치를 저장할 변수 left, right를 선언하고 각각 *와 #을 넣어준다.
  • 좌표간의 거리를 계산하는 함수 distance를 cur, left_l, right_l을 인자로 갖도록 선언한다.
    -> 키패드를 표현할 배열 pad를 선언하고 키패드를 2차원 배열로 구현한다.
    -> 검사할 숫자의 x, y좌표를 저장할 변수 cur_x, cur_y와 왼손의 x, y좌표를 저장할 변수 left_x, left_y와 오른손의 x, y좌표를 저장할 변수 right_x, right_y를 0으로 선언한다.
    -> 4번 반복하는 i에 대한 for문을 돌린다.
    --> pad[i]에 cur이 있을 경우 cur_x에 i를, cur_y에 pad[i]에서 cur의 인덱스를 저장한다.
    --> pad[i]에 left_l이 있을 경우 left_x에 i를, left_y에 pad[i]에서 left_l의 인덱스를 저장한다.
    --> pad[i]에 right_l이 있을 경우 right_x에 i를, right_y에 pad[i]에서 right_l의 인덱스를 저장한다.
    -> 왼손과 누르려는 숫자의 거리를 left_x-cur_x의 절댓값 + left_y-cur_y의 절댓값으로 구해 left_dis에 저장한다.
    -> 오른손과 누르려는 숫자의 거리를 right_x-cur_x의 절댓값 + right_y-cur_y의 절댓값으로 구해 right_dis에 저장한다.
    -> 만약 left_dis보다 right_dis가 더 크다면 'L'을 반환한다.
    -> 만약 left_dis보다 right_dis가 더 작다면 'R'을 반환한다.
    -> left_dis와 right_dis가 같을 경우,
    --> hand가 left라면 'L'을 반환한다.
    --> hand가 right라면 'R'을 반환한다.
  • dn이 존재하는 동안 반복하는 while문을 돌린다.
    -> dn을 popleft한 값을 임시 변수 tmp에 저장한다.
    -> 만약 tmp가 l에 포함될 경우,
    --> answer에 'L'을 추가한다.
    --> left를 tmp로 갱신한다.
    -> 만약 tmp가 r에 포함될 경우,
    --> answer에 'R'을 추가한다.
    --> right를 tmp로 갱신한다.
    -> 그 외의 경우,
    --> answer에 distance(tmp, left, right) 함수를 호출하여 반환되는 값을 추가한다.
    --> 만약 방금 answer에 추가된 answer[-1]이 'L'일 경우,
    ---> left를 tmp로 갱신한다.
    --> 만약 방금 answer에 추가된 answer[-1]이 'R'일 경우,
    ---> right를 tmp로 갱신한다.
  • answer를 반환한다.

solution.py

from collections import deque
def solution(numbers, hand):
    answer = ''
    dn=deque(numbers)
    l=['1', '4', '7']
    r=['3', '6', '9']
    left, right = '*', '#'
    def distance(cur, left_l, right_l):
        pad=[['1','2','3'], ['4','5','6'],['7','8','9'],['*', '0', '#']]
        cur_x, cur_y, left_x, left_y, right_x, right_y = 0, 0, 0, 0, 0, 0
        for i in range(4):
            if pad[i].count(cur)==1:
                cur_x, cur_y = i, pad[i].index(cur)
            if pad[i].count(left_l)==1:
                left_x, left_y = i, pad[i].index(left_l)
            if pad[i].count(right_l)==1:
                right_x, right_y = i, pad[i].index(right_l)
        left_dis=abs(left_x-cur_x)+abs(left_y-cur_y)
        right_dis=abs(right_x-cur_x)+abs(right_y-cur_y)
        if left_dis<right_dis:
            return 'L'
        elif left_dis>right_dis:
            return 'R'
        else:
            if hand=='left':
                return 'L'
            else:
                return 'R'
    while dn:
        tmp=str(dn.popleft())
        if tmp in l:
            answer+='L'
            left=tmp
        elif tmp in r:
            answer+='R'
            right=tmp
        else:
            answer+=distance(tmp, left, right)
            if answer[-1]=='L':
                left=tmp
            else:
                right=tmp
    return answer

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글