[アルゴリズム]キーボードを押す

プログラマー2020ココア実習生コードテストにおける問題

ぶんせき

1.携帯電話の配列を実現

phone = [[3,1],[0,0],[0,1],[0,2],[1,0],[1,1],[1,2],[2,0],[2,1],[2,2]]

2.中央欄をクリックして手を近づける

左手と番号を押す距離、右手と番号を押す距離を測定します.
左手は押す番号と2の距離、右手は押す番号と2の距離です.

コード#コード#

def check(pos1, pos2):
    return abs(pos1[0]-pos2[0])+abs(pos1[1]-pos2[1])

def solution(numbers, hand):
    answer = ''
    phone = [[3,1],[0,0],[0,1],[0,2],[1,0],[1,1],[1,2],[2,0],[2,1],[2,2]]
    left, right = [3,0],[3,2]
    for i in numbers:
        if i in [1,4,7]:
            left = phone[i]
            answer+='L'
        elif i in [3,6,9]:
            right = phone[i]
            answer+='R'
        elif i in [2,5,8,0]:
            leftCnt = check(left, phone[i])
            rightCnt = check(right, phone[i])
            if leftCnt>rightCnt:
                right = phone[i]
                answer+='R'
            elif leftCnt<rightCnt:
                left = phone[i]
                answer+='L'
            else:
                if hand=="right":
                    right = phone[i]
                    answer+='R'
                else:
                    left = phone[i]
                    answer+='L'
    return answer