[python]키패드 누르기

eunvi·2021년 11월 30일
0

프로그래머스

목록 보기
4/25
post-thumbnail

2020 카카오 인턴십>키패드 누르기

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

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

  • 엄지손가락은 상하좌우 4가지 방향으로만 이동할 수 있으며 키패드 이동 한 칸은 거리로 1에 해당합니다.
  • 왼쪽 열의 3개의 숫자 1, 4, 7을 입력할 때는 왼손 엄지손가락을 사용합니다.
  • 오른쪽 열의 3개의 숫자 3, 6, 9를 입력할 때는 오른손 엄지손가락을 사용합니다.
  • 가운데 열의 4개의 숫자 2, 5, 8, 0을 입력할 때는 두 엄지손가락의 현재 키패드의 위치에서 더 가까운 엄지손가락을 사용합니다.
    • 만약 두 엄지손가락의 거리가 같다면, 오른손잡이는 오른손 엄지손가락, 왼손잡이는 왼손 엄지손가락을 사용합니다.

순서대로 누를 번호가 담긴 배열 numbers, 왼손잡이인지 오른손잡이인 지를 나타내는 문자열 hand가 매개변수로 주어질 때, 각 번호를 누른 엄지손가락이 왼손인 지 오른손인 지를 나타내는 연속된 문자열 형태로 return 하도록 solution 함수를 완성해주세요.

[제한사항]

  • numbers 배열의 크기는 1 이상 1,000 이하입니다.
  • numbers 배열 원소의 값은 0 이상 9 이하인 정수입니다.
  • hand는 "left" 또는 "right" 입니다.
    • "left"는 왼손잡이, "right"는 오른손잡이를 의미합니다.
  • 왼손 엄지손가락을 사용한 경우는 L, 오른손 엄지손가락을 사용한 경우는 R을 순서대로 이어붙여 문자열 형태로 return 해주세요.

[입출력 예]

나의 풀이

생각해보기!

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

참고하세요!

  • 키패드를 key(숫자),value(행렬) 딕셔너리 형태로 나타냈다.

다른 사람의 풀이 #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

참고하세요!

  • 2차원 리스트로 12*10 행렬을 표현했다.
  • p[1][2]=1 ➡ 키패드 1과 2사이의 거리가 1임을 뜻한다.
profile
please study

0개의 댓글

Powered by GraphCDN, the GraphQL CDN