[알고리즘] 프로그래머스 - 조이스틱 - JAVA

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

입출력 예

name	return
"JEROEN"	56
"JAN"	23

---

풀이과정

조이스틱을 상하로 움직이는 경우는 아스키코드를 이용하면 쉽게 계산할 수 있다.

 

이 문제에서 어려웠던 점은 조이스틱을 좌우로 움직이는 경우를 생각하는 것이다.

 

조이스틱을 좌우로 움직이는 경우의 최소값은 두가지 경우로 나뉜다.

 

1. 오른쪽으로 쭉 가는 경우 (연속된 A가 적을때)

2. 오른쪽으로 가다가 왼쪽으로 돌아가야 하는 경우 (중간에 연속된 A가 있어서 방문할 필요가 없는 경우)

 

1의 경우는 단순하게 name.length() - 1을 하면 구할 수 있지만, 2의 경우를 이해하는데 어려움을 겪었다.

 

이해를 위해 단순하게 생각해보자.

 

JANAAAAAAKB라는 문자열이 있다면 결국 N ~ K로 이동하는데 필요한 횟수를 구하면 된다.

 

name.length() - 1과 N에서 왼쪽으로 거꾸로가서 K까지 이동하는데 필요한 카운트를 비교하여 더 적은 값이 조이스틱 이동에 필요한 최소값이다. 

 

이것만 이해한다면 문제를 풀 수 있을 것이다. 주석을 보며 이해해보자.

 

소스코드

import java.util.*;

class Solution {
    public int solution(String name) {
        int answer = 0;
        
        // 오른쪽으로 쭉 가는 경우
        int cost = name.length() - 1;
        
        for (int i = 0; i < name.length(); i++) {
            char dst = name.charAt(i);
            int count = 0;
    
            if (dst != 'A') {
                count = min(dst);
            }
            answer += count;
            
            // 왼쪽으로 돌아가는 경우
            int nextIndex = i + 1;
           
            // 연속된 A가 끝난 인덱스가 nextIndex
            while (nextIndex < name.length() && name.charAt(nextIndex) == 'A')
              nextIndex++;
            
            // i * 2를 통해 0번 인덱스까지 돌아가는 횟수를 구한다.
            // 그리고 name.length()에서 nextIndex를 빼 0에서 뒤로 가는데 필요한 횟수를 구한다.
            cost = Math.min(cost, (i * 2) + name.length() - nextIndex);
        }
        
        answer += cost;
        return answer;
    }
    
    public int min(char dst) {
        return Math.min(dst - 'A', ('Z' - 'A' + 1) - (dst - 'A'));
    }
}