[프로그래머스-Lv2] 조이스틱 / Python

코딩테스트 연습 - 조이스틱

import sys

input = sys.stdin.readline

name = input().rstrip()


def solution(name):
    change = [min(ord(i) - ord("A"), ord("Z") - ord(i) + 1) for i in name]
    answer, idx = 0, 0
    while True:
        answer += change[idx]
        change[idx] = 0
        if sum(change) == 0:
            return answer

        left, right = 1, 1
        while change[idx - left] == 0:
            left += 1
        while change[idx + right] == 0:
            right += 1
        answer += left if left < right else right
        idx += -left if left < right else right
    return answer


print(solution(name))

 

해설

change에 상 이동과 하 이동 중 최소를 넣어준다. 이 때 ord를 사용해서 실질적인 거리를 구할 수 있도록 한다.

상하 이동거리를 answer에 축적시킨다.

좌우 이동거리를 구하고, 더 짧은 방향으로 이동한다. 이동한 거리만큼 answer에 축적시킨다.

 

좌로 이동하는 거리가 더 짧으면 answer에 좌로 이동하는 거리만큼 축적.

우로 이동하는 거리가 더 짧으면 answer에 우로 이동하는 거리만큼 축적.

 

좌로 이동하는 거리가 더 짧으면 idx는 현재 idx에 -left를 해준다.

우로 이동하는 거리가 더 짧으면 idx는 현재 idx에 +right를 해준다.