[코딩테스트] 2020년 카카오 - 숫자 키패드 문제

kysung95·2021년 5월 3일

코딩테스트

목록 보기

1/10

안녕하세요. 김용성입니다.
오늘은 2020년에 카카오 인턴십 코딩테스트에서 기출되었던 '숫자키패드' 문제를 풀어보고자 합니다.

문제

2020년 카카오 인턴십 1번 문제 링크

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

엄지손가락은 상하좌우 4가지 방향으로만 이동할 수 있으며 키패드 이동 한 칸은 거리로 1에 해당합니다.
왼쪽 열의 3개의 숫자 1, 4, 7을 입력할 때는 왼손 엄지손가락을 사용합니다.
오른쪽 열의 3개의 숫자 3, 6, 9를 입력할 때는 오른손 엄지손가락을 사용합니다.
가운데 열의 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 해주세요.

풀이

1번으로 출제된 문제인 만큼 어려운 문제는 아니였지만, 귀찮은 문제였다고 생각합니다.
저는 파이썬 언어에 대한 숙련도가 그리 높지 않아서 조건문을 최대한 활용해서 문제에 접근했어요.

이차원 리스트로 키패드 변환

저는 키패드를 3x2 이차원 리스트로 생각하고 문제풀이를 시작하였습니다.
따라서 array 리스트를 생성해 준 뒤 다음과 같이 키패드를 할당해준 후,
오른손과 왼손이 각각 '#'과 '*'에서 시작한다고 했으므로 right와 left 또한 행렬 리스트로 초기화를 시켜주었습니다.

 array=[[1,2,3],[4,5,6],[7,8,9],[10,0,12]]
 right=[3,2]
 left=[3,0]

영역별 조건문 분기

키보드를 살펴보면 3으로 나누어떨어지는 3,6,9 라인은 무조건 오른손을 써서 키패드를 누를것이고,
3으로 나누었을 때 나머지가 1이 되는 1,4,7 라인은 무조건 왼손을 써서 키패드를 누를 거예요.
따라서 다음과 같이 분기를 했습니다.

if numbers[i]!=0 and numbers[i]%3==0 or numbers[i]%3==1:


else:

참고로 number가 0이 아닌 경우를 집어넣어주는 이유는 0%3 또한 나머지가 0으로 나오기 때문이예요. 저 부분은 반드시 처리를 해주어야 합니다.

가운데 2,5,8,0 라인 처리

이 부분에서 시간이 조금 오래걸렸어요. 문제 조건을 잘 읽어야하는 부분입니다.
저 같은 경우에는 처음에는 키패드간의 거리라고 생각을 했는데 그게 아니라 키패드를 움직이는 횟수라는 점.. 그걸 캐치하지 못하고 sqrt를 사용해서 풀었다가 자꾸 13~20 번 테스트 케이스에서 실패했다고 나오더라구요..

이게 무슨 말이냐면 오른 손이 6에 있고 왼손이 8에 있는 상황에서 sqrt를 사용한다면 당연히 피타고라스 정리로 인해 2~6의 거리는 1.414, 2에서 8까지의 거리는 2 이렇게 나오기 때문에 6에 있던 오른 손이 2번 키패드를 누르게 될 거라고 판단할텐데요. 그것이 아니라 8이나 6이나 둘 다 똑같이 2까지는 두칸이기 때문에 같게 처리가 된다는 점이예요. 저처럼 시간을 허비하기 싫으신 분들은 꼭 문제를 잘 읽고 머릿속에서 시뮬레이션을 돌리시길 바랍니다 ㅠㅠ

변경 전 코드 (뻘짓)

right_distance=math.sqrt((right_x*right_x)+(right_y*right_y))
left_distance=math.sqrt((left_x*left_x)+(left_y*left_y))

변경 후 코드

right_distance=abs(right_x)+abs(right_y)
left_distance=abs(left_x)+abs(left_y)

전체적인 FLOW

저는 주어진 numbers 배열의 길이만큼 for문을 돌리면서 분기시킨 조건문대로 그냥 돌려주기만 했어요. 다만 키패드를 어떤 손으로 눌러줄지 결정될 때마다 처음에 선언해준 right,left 리스트에 해당 키패드의 좌표값을 넣어주었고요. 그렇게 현재 오른손 엄지손가락이 어디에 있는지, 왼손 엄지손가락이 어디있는지를 생각해놓으며 문제를 해결했습니다.

다만 오른손으로 키패드를 눌러준 뒤에 right 위치를 변환시켜줄 때 [(numbers[i]//3),2]이 아닌 [(numbers[i]//3)-1,2]로 해주어야해요. 그 이유는 3의 배수 3,6,9 라인은 //3을 해준 값이 1에서 시작하기 때문이죠.

최종 코드

최종 코드는 다음과 같습니다.


# 1번 문제

def solution(numbers, hand):
    
    answer=""
      
    array=[[1,2,3],[4,5,6],[7,8,9],[10,0,12]]
    right=[3,2]
    left=[3,0]
    
    for i in range(len(numbers)):
        if numbers[i]!=0 and numbers[i]%3==0 or numbers[i]%3==1:
            if numbers[i]%3==0:
                answer+="R"
                right=[(numbers[i]//3)-1,2]
            else:
                answer+="L"
                left=[numbers[i]//3,0]
        else:
            if numbers[i]==0:
                current_key=[3,1]
            else:
                current_key=[numbers[i]//3,1]
                
            right_x=right[0]-current_key[0]
            right_y=right[1]-current_key[1]
            left_x=left[0]-current_key[0]
            left_y=left[1]-current_key[1]
            right_distance=abs(right_x)+abs(right_y)
            left_distance=abs(left_x)+abs(left_y)

            if right_distance>left_distance:
                answer+="L"
                left=current_key
            elif right_distance==left_distance:
                if hand=="right":
                    answer+="R"
                    right=current_key
                else:
                    answer+="L"
                    left=current_key
            else:
                answer+="R"
                right=current_key      
      
    return answer