[카카오 인턴] 키패드 누르기

programmers.co.kr/learn/courses/30/lessons/67256?language=python3

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

def search(key_pad, now_num):

    for i in range(len(key_pad)):
        for j in range(len(key_pad[i])):
            if now_num == key_pad[i][j]:
                a= [i,j]
    return a
def solution(numbers, hand):
    answer = ''
    key_pad = [[1,2,3],
               [4,5,6],
               [7,8,9],
               ["*",0,"#"]]

    l_now = key_pad[3][0]
    r_now = key_pad[3][2]
    for i in numbers:
        a = search(key_pad, l_now)
        b = search(key_pad, r_now)
        c = search(key_pad,i)
        if i == 1 or i == 4 or i == 7:
            answer += "L"
            l_now = key_pad[c[0]][c[1]]
        elif i == 3 or i == 6 or i == 9:
            answer += "R"
            r_now = key_pad[c[0]][c[1]]
        elif i == 2 or i == 5 or i == 8 or i == 0:
            l_length = abs(c[0]-a[0]) + abs(c[1]-a[1])
            r_length = abs(b[0]-c[0]) + abs(b[1]-c[1])

            if l_length == r_length and hand == "right":
                answer += "R"
                r_now = key_pad[c[0]][c[1]]
            elif l_length == r_length and hand == "left":
                answer += "L"
                l_now = key_pad[c[0]][c[1]]
            elif l_length > r_length:
                answer += "R"
                r_now = key_pad[c[0]][c[1]]
            elif l_length < r_length:
                answer += "L"
                l_now = key_pad[c[0]][c[1]]
    return answer

문제가 어렵지는 않았다.

문제를 잘 읽자