프로그래머스: 조이스틱

문제

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

 

문제 풀이

 조이스틱으로 알파벳 이름을 완성하고자 한다. 이때 조이스틱을 움직이면 다음과 같이 동작한다.

 

• ▲ - 다음 알파벳
• ▼ - 이전 알파벳
  • (A에서 아래쪽으로 이동하면 Z로)
• ◀ - 커서를 왼쪽으로 이동
  • (첫 번째 위치에서 왼쪽으로 이동하면 마지막 문자에 커서)
• ▶ - 커서를 오른쪽으로 이동

 문제를 풀기 위해서는 각 알파벳이 A에서 이동하는 것이 빠른지, Z에서 이동하는 것이 빠른지에 따라 초기에 조이스틱을 이동해야 하는 횟수를 가진 리스트를 초기화한다. 그리고 첫 문자는 A로 시작하므로 dist의 합이 0인 경우는 조이스틱을 움직일 필요가 없다. 이외에는 현재 문자의 왼쪽이나 오른쪽 중 더 빨리 조작할 수 있는 것을 우선적으로 처리하면 문제에서 구하고자 하는 답을 찾을 수 있다.

 

코드

def solution(name):
    dist = [min(ord(s) - ord("A"), ord("Z") - ord(s) + 1) for s in name]

    idx, answer = 0, 0
    while True:
        answer += dist[idx]
        dist[idx] = 0
        
        if sum(dist) == 0:
            break
            
        left, right = 1, 1
        
        while not dist[idx - left]:
            left += 1
            
        while not dist[idx + right]:
            right += 1
            
        if left >= right:
            idx += right
            answer += right
        else:
            idx -= left
            answer += left
    
    return answer