[코딩 테스트 연습(파이썬/Python)] 프로그래머스 _ 조이스틱

https://programmers.co.kr/learn/courses/30/lessons/42860

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

[문제 설명]

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

 

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

▲ - 다음 알파벳

▼ - 이전 알파벳 (A에서 아래쪽으로 이동하면 Z로)

◀ - 커서를 왼쪽으로 이동 (첫 번째 위치에서 왼쪽으로 이동하면 마지막 문자에 커서)

▶ - 커서를 오른쪽으로 이동

 

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

- 첫 번째 위치에서 조이스틱을 위로 9번 조작하여 J를 완성합니다.

- 조이스틱을 왼쪽으로 1번 조작하여 커서를 마지막 문자 위치로 이동시킵니다.

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

 

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

 

[제한 조건]

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

고생했다.. A가 없는 경우야 쉽고, 결국 A가 어디에 얼만큼 위치하느냐가 중요하다. 출발 지점을 기준으로 양 옆에 연속된 A가 있는 경우는 쉽고, 중간에 연속된 A가 있는 경우에는 쭉 한 방향 진행이 나을지, 앞으로 갔다가 뒤로 가는 것이 나은지 비교해야한다. (이때 왼쪽 커서만이 오른쪽 끝으로 이동할 수 있고, 반대로 오른쪽 커서로 다시 왼쪽 끝으로 오는 것은 불가능함을 문제에서 잘 확인해야 한다.)

# 정답
def solution(name):
    alpha = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O',
             'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
    import re
    tmp = 0 ; cnt1 = 0
    for x in name:
        tmp += min(alpha.index(x) , 26 - alpha.index(x))
    pattern = re.compile('[A]+')
    temp = re.findall(pattern, name[1:])
    if temp:
        aa = name[1:].split(max(temp))
        # 시작점 기준 양쪽 끝에 A가 최대로 연속된 부분이 존재하는 경우
        if '' in aa:
            cnt1 = len(max(temp))
        # 시작점 기준 양쪽 끝이 아닌 가운데에 A가 최대로 연속된 부분이 존재하는 경우,
        # 앞으로 갔다가 왼쪽으로 진행하는 것 vs 한쪽 방향으로 쭉 진행하는 것 사이에 비교 필요
        # ex. 'AXAADAAAAAAAADA'
        else:
            cnt2 = 2*len(aa[0]) + len(aa[-1])
            cnt1 = min (cnt2, len(name) - 1 )
            return tmp + cnt1
    return tmp + len(name) - 1 - cnt1

예를 들어 'AXAADAAAAAAAADA'의 경우, 4번 인덱스의 자리까지 이동해 D로 바꾸고, 오는 길에 1번 인덱스를 바꾸고, 왼쪽 키를 이용해 오른쪽 끝으로 이동한 뒤 -2번 인덱스를 D로 바꾸는 것이 최단 경로이고, 이를 위해 else구문이 필요하다. else 조건에 해당하더라도, 'AXAADAAADA' 와 같은 경우 쭉 한 방향 진행이 더 경로가 짧다.