[프로그래머스] Level2, 조이스틱 (Python)

문제

https://programmers.co.kr/learn/courses/30/lessons/42860?language=python3# 

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

 

코드

def solution(name):
    answer = 0
    min_name = [min(ord(i) - ord('A'), ord('Z') - ord(i) + 1) for i in name]
    print(min_name)
    idx = 0
    while True:
        answer += min_name[idx]
        min_name[idx] = 0
        if sum(min_name) == 0:
            break
        
        left,right = 1,1
        while min_name[idx - left] == 0:
            left += 1
        while min_name[idx + right] == 0:
            right += 1
        
        answer += left if left < right else right
        idx += -left if left < right else right
        
    return answer

조이스틱에서 위로 움직여서 알파벳을 바꾸는 것과 아래로 움직여서 바꾸는 것중에 최솟값을 찾아 리스트로 만든다.

ord로 문자열을 아스키코드로 만들어서 뺴주면 몇번 이동하는지 구할 수 있다.

 

그 다음에는 조이스틱을 왼쪽 또는 오른쪽으로 움직여야할 지를 판단해야하는데 A일 경우에는 바꾸지 않고 넘어가도 되므로 A가 right쪽에 많은지 left쪽에 많은지 판단해 더 적은 쪽으로 idx를 움직이게 한다.