[프로그래머스] 조이스틱 - python

조이스틱

문제 설명

조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 A로만 이루어져 있습니다. ex) 완성해야 하는 이름이 세 글자면 AAA, 네 글자면 AAAA

조이스틱을 각 방향으로 움직이면 아래와 같습니다.

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

예를 들어 아래의 방법으로 "JAZ"를 만들 수 있습니다.

- 첫 번째 위치에서 조이스틱을 위로 9번 조작하여 J를 완성합니다.
- 조이스틱을 왼쪽으로 1번 조작하여 커서를 마지막 문자 위치로 이동시킵니다.
- 마지막 위치에서 조이스틱을 아래로 1번 조작하여 Z를 완성합니다.
따라서 11번 이동시켜 "JAZ"를 만들 수 있고, 이때가 최소 이동입니다.

만들고자 하는 이름 name이 매개변수로 주어질 때, 이름에 대해 조이스틱 조작 횟수의 최솟값을 return 하도록 solution 함수를 만드세요.

제한 사항

• name은 알파벳 대문자로만 이루어져 있습니다.
• name의 길이는 1 이상 20 이하입니다.

풀이

각 문자마다 두 개의 최적해 선택해야한다.

• 알파벳을 고르는 최소 값
• 커서를 이동 시키는 최소값

다음 문자가 A가 아닌 위치를 찾아서 오른쪽으로 가는 것과 왼쪽으로 가는 것 중에 어떤 게 더 빠른지 비교하는 방법도 있지만 몇몇 예외 케이스를 통과하지 못했다.

입력값 〉 "ABABAAAAABA"
기댓값 〉 11
​
입력값 〉 "BBBAAAB"
기댓값 〉 9

다른 사람 풀이를 이것 저것 봤는데, 그냥 문자를 하나씩 구현하면서 매번 왼쪽으로 이동했을 때, 오른쪽으로 이동했을 때를 카운트해서 비교해주는 완전탐색이 제일 정확한 것 같다.

def solution(name):
    answer = 0
    n = len(name)
    
    # name의 길이만큼 A 배열 만들기
    default = ['A' for _ in range(n)]
    
    i = 0
    while True:
       # 알파벳 최솟값 계산
        answer += min(ord('Z') - ord(name[i]) + 1, ord(name[i]) - ord('A'))
        
        #해당 알파벳으로 해당 위치 바꿔줌
        default[i] = name[i]
        
        #만약에 문자열이 완성되었다면 종료
        if ''.join(default) == name:
            return answer
        
        # 현재 위치에서 왼쪽으로 이동해서 다음 위치 찾을 경우 카운트
        left = i
        left_cnt = 0
        while default[left] == name[left]:
            left = (left - 1) % n
            left_cnt += 1
        
        # 현재 위치에서 오른쪽으로 이동해서 다음 위치 찾을 경우 카운트
        right = i
        right_cnt = 0
        while default[right] == name[right]:
            right = (right + 1) % n
            right_cnt += 1
        
        # 둘 중에 더 작은 값으로 현재 인덱스 바꿔주고 
        # 카운트한 값 answer에 더하기
        if left_cnt < right_cnt:
            i = left
            answer += left_cnt
        else:
            i = right
            answer += right_cnt
    return answer