[프로그래머스] 조이스틱 [탐욕법]

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 이하입니다.

입출력 예

name	return
"JEROEN"	56
"JAN"	23

풀이

특정 알파벳을 빠르게 만드는 법, 단어를 만들기 위해 적게 움직이는 법 두 가지를 고려해야 하는 문제입니다.
먼저, 특정 알파벳을 만드는 방법은 ABC 순, AZYX 순이 있습니다. 이는 어떤 문자의 ascii 코드가 A에 가까운지 Z에 가까운지에 따라 결정되므로 뺄셈을 통해 간단히 구할 수 있습니다.

# 어떤 문자 x에 대한 최단 거리
min(ord(x)-ord('A'), ord('Z')-ord(x)+1)

이를 응용하면 조이스틱을 위아래로 움직이는 최소 횟수를 쉽게 구할 수 있습니다.

answer = sum([min(ord(x)-ord('A'), ord('Z')-ord(x)+1) for x in name])

이제 바꿀 알파벳으로 이동하기 위해 좌우로 움직이는 최소 횟수를 구해야 합니다. 이를 구하기 위해선, A가 아닌 가장 가까운 알파벳으로 이동하면서 횟수를 세어주면 됩니다. 이때, 문자열이 원형으로 이어져있다고 생각하며 이 과정을 A가 아닌 문자의 수만큼 반복해줍니다.

BBAAAABA를 예로 들어볼까요? BBAAAABA에 대한 기댓값은 7입니다. 조이스틱 방향으로 표현하면 다음과 같습니다. ▲▶▲◀◀◀▲
이제 앞서 설명한 규칙과 일치하는지 알아봅시다.

pos	name[pos]	i	pos+i	name[pos+i]	pos-i	prev
0	B	1	1	B	-1	A	왼쪽으로 1만큼 이동
1	B	1	2	A	0	(visited)
		2	3	A	-1	A
		3	4	A	-2	B	오른쪽으로 3만큼 이동

def solution(name):
	# 주어진 이름에서 A가 아닌 문자의 수
    num = sum([1 if x!='A' else 0 for x in name])
    # 모두 A인 경우 (e.g., AAA)
    if not num:
        return 0
        
    # 주어진 문자에 대한 위아래 이동의 총합
    answer = sum([min(ord(x)-ord('A'), ord('Z')-ord(x)+1) for x in name])
    
    # item assignment를 위해 str에서 list로 변환
    name = list(name)

    # 인덱스 0의 경우 좌우로 이동하지 않아도 A로 만들 수 있다.
    if name[0] != "A":
        num -= 1
        name[0] = "A"
        
	# 최초 인덱스 위치는 0
    pos = 0
    # 남은 문자의 수만큼 반복
    for _ in range(num):
        for i in range(1, len(name)):
            # 원형 리스트에서 오른쪽으로 i만큼 이동한 인덱스
            r = (pos+i) % len(name)
            # 원형 리스트에서 왼쪽으로 i만큼 이동한 인덱스
            l = (pos-i) % len(name)
            
            if name[r] != "A":
                answer += i
                pos = r
                name[pos] = "A"
                break

            if name[l] != "A":
                answer += i
                pos = l
                name[l] = "A"
                break
    return answer