[99클럽 코테 스터디 13일차 TIL] Joystick 문제 풀이

문제

프로그래머스 - 조이스틱 문제를 보고 풀이한 내용이다.

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

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

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

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

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

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

풀이

각 예제의 정답을 찾는 과정을 생각해보니 위 그림과 같았다.

다음 알파벳으로 넘어가는 ▲나 ▼의 경우 아스키 코드를 이용해 더 가까운 쪽을 판별할 수 있을 것 같았다.

그런데 ◀▶ 판별은 쉽지 않았다. 예를 들면 BAAB의 경우 오른쪽으로만 이동하면 5번의 조작 횟수가 필요하지만, 왼쪽으로 먼저 이동하면 A의 경우는 조작할 필요가 없으므로 3번의 조작 횟수로 원하는 문자를 만들 수 있었다.

따라서 처음에는 조건을 다음과 같이 생각했다.

1. ▲▼ 판별하기
  - (완성해야할 문자 - A)와 (Z - 완성해야할 문자 + 1) 중 작은 것을 선택해서 전자라면 ▲, 후자라면 ▼

2. ◀▶ 판별하기
  - 그냥 왼쪽, 오른쪽 둘 다 가보고 더 짧은 경로 선택

public int solution(String name) {
        int rightMoveCount = 0;
        int leftMoveCount = 1;
        String[] nameArr = name.split("");

        // 조이스틱을 오른쪽으로 움직이는 경우
        for (int idx = 0; idx < nameArr.length; idx++) {
            // A인 경우 오른쪽으로 한 칸 움직인다
            if (nameArr[idx].equals("A")) {
                rightMoveCount += 1;
                continue;
            }
            // 다른 문자인 경우 해당 문자로 만들어주고 한 칸 움직인다
            rightMoveCount += determineSmall(nameArr[idx].charAt(0)) + 1;
        }

        // 조이스틱을 왼쪽으로 움직이는 경우
        for (int idx = nameArr.length - 1; idx >= 0; idx--) {
            // A인 경우 왼쪽으로 한 칸 움직인다
            if (nameArr[idx].equals("A")) {
                leftMoveCount += 1;
                continue;
            }
            leftMoveCount += determineSmall(nameArr[idx].charAt(0)) + 1;
        }

        return leftMoveCount <= rightMoveCount ? leftMoveCount : rightMoveCount;
    }

    // ▲로 움직였을 때와 ▼로 움직였을 때 중 더 작은 값을 선택
    public int determineSmall(char spelling) {
        int smallNumber = 0;
        // ▲로 움직였을 때 더 작은 경우
        if (spelling - 'A' < (('Z' - spelling) + 1)) {
            smallNumber = spelling - 'A';
        } else { // ▼로 움직였을 때 더 작은 경우
            smallNumber = 'Z' - spelling + 1;
        }
        return smallNumber;
    }

그런데 내가 간과한 점이 있었다. ◀▶ 조작의 경우 한 방향만을 선택해서 가는 게 아니라 오른쪽으로 갔다가 다시 왼쪽으로 돌아올 수도 있고, 왼쪽으로 갔다가 다시 오른쪽으로 조작할 수도 있었다.

정답 코드

public int solution(String name) {
        int answer = 0;
        int move = name.length() - 1; // 오른쪽으로 쭉 간 횟수

        for(int i = 0; i < name.length(); i++) {
            answer += Math.min(name.charAt(i) - 'A', 26 - (name.charAt(i) - 'A'));
            if (i < name.length() - 1 && name.charAt(i + 1) == 'A') {
                int endA = i + 1;
                while(endA < name.length() && name.charAt(endA) == 'A')
                    endA++;
                move = Math.min(move, i * 2 + (name.length() - endA));
                move = Math.min(move, i + (name.length() - endA) * 2);
            }
        }
        return answer + move;
    }