[문제 설명]
스마트폰 전화 키패드의 각 칸에 다음과 같이 숫자들이 적혀 있습니다.
이 전화 키패드에서 왼손과 오른손의 엄지손가락만을 이용해서 숫자만을 입력하려고 합니다.
맨 처음 왼손 엄지손가락은 * 키패드에 오른손 엄지손가락은 # 키패드 위치에서 시작하며, 엄지손가락을 사용하는 규칙은 다음과 같습니다.
순서대로 누를 번호가 담긴 배열 numbers, 왼손잡이인지 오른손잡이인 지를 나타내는 문자열 hand가 매개변수로 주어질 때, 각 번호를 누른 엄지손가락이 왼손인 지 오른손인 지를 나타내는 연속된 문자열 형태로 return 하도록 solution 함수를 완성해주세요.
[제한사항]
[입출력 예]
나의 풀이
def solution(numbers, hand):
answer = ''
# 키패드를 행렬로 나타내기
keypad = [[4,2],[1,1],[1,2],[1,3],[2,1],[2,2],[2,3],[3,1],[3,2],[3,3]]
left_number = [4,1] # 왼손 엄지손가락 시작점
right_number = [4,3] # 오른손 엄지손가락 시작점
for i in numbers:
if i in [1,4,7]: # 왼손 엄지손가락을 사용하는 열
answer += 'L'
left_number = keypad[i]
elif i in [3,6,9]: # 오른손 엄지손가락을 사용하는 열
answer += 'R'
right_number = keypad[i]
else: # 가운데 열의 4개의 숫자를 입력할 때
if answer == '': # 처음 입력할 경우
if hand == 'left':
answer += 'L'
left_number = keypad[i]
else:
answer += 'R'
right_number = keypad[i]
else: # 앞서 입력이 있었을 경우
# 두 엄지손가락의 현재 키패드의 위치에서 더 가까운 엄지손가락을 사용
# 행렬로 나타낸 키패드 간의 거리를 절댓값의 합으로 표현
left_dist = sum([abs(x-y) for x,y in zip(keypad[i],left_number)])
right_dist = sum([abs(x-y) for x,y in zip(keypad[i],right_number)])
if left_dist < right_dist: # 왼손 엄지가 가까울 때
answer += 'L'
left_number = keypad[i]
elif left_dist > right_dist: # 오른손 엄지가 가까울 때
answer += 'R'
right_number = keypad[i]
else: # 두 손 엄지의 거리가 같을 때
if hand == 'left':
answer += 'L'
left_number = keypad[i]
else:
answer += 'R'
right_number = keypad[i]
return answer
키 패드를 행렬로 나타내고 키 패드 간의 거리를 표현하기에서 조금 헤맸다. 다음에 비슷한 문제는 더 수월하게 풀 수 있을 거 같다.🙂
다른 사람의 풀이 #1
def solution(numbers, hand):
answer = ''
key_dict = {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)}
left = [1,4,7]
right = [3,6,9]
lhand = '*'
rhand = '#'
for i in numbers:
if i in left:
answer += 'L'
lhand = i
elif i in right:
answer += 'R'
rhand = i
else:
curPos = key_dict[i]
lPos = key_dict[lhand]
rPos = key_dict[rhand]
ldist = abs(curPos[0]-lPos[0]) + abs(curPos[1]-lPos[1])
rdist = abs(curPos[0]-rPos[0]) + abs(curPos[1]-rPos[1])
if ldist < rdist:
answer += 'L'
lhand = i
elif ldist > rdist:
answer += 'R'
rhand = i
else:
if hand == 'left':
answer += 'L'
lhand = i
else:
answer += 'R'
rhand = i
return answer
다른 사람의 풀이 #2
def solution(numbers, hand):
l = 10
r = 11
answer = ""
p = [[0, 4, 3, 4, 3, 2, 3, 2, 1, 2],
[4, 0, 1, 2, 0, 2, 3, 0, 3, 4],
[3, 1, 0, 1, 2, 1, 2, 3, 2, 3],
[4, 2, 1, 0, 3, 2, 1, 4, 3, 2],
[3, 0, 2, 3, 0, 1, 2, 0, 2, 3],
[2, 2, 1, 2, 1, 0, 1, 2, 1, 2],
[3, 3, 2, 1, 2, 1, 0, 3, 2, 1],
[2, 0, 3, 4, 0, 2, 3, 0, 1, 2],
[1, 3, 2, 3, 2, 1, 2, 1, 0, 1],
[2, 4, 3, 2, 3, 2, 1, 2, 1, 0],
[1, 0, 4, 5, 0, 3, 4, 0, 2, 3],
[1, 5, 4, 0, 4, 3, 0, 3, 2, 0]]
for i in numbers:
if i in [1, 4, 7]:
l = i
answer += "L"
elif i in [3, 6, 9]:
r = i
answer += "R"
else:
if p[l][i] < p[r][i]:
l = i
answer += "L"
elif p[l][i] > p[r][i]:
r = i
answer += "R"
elif hand == "left":
l = i
answer += "L"
else:
r = i
answer += "R"
return answer