알고리즘 공부/탐욕알고리즘(Greedy)

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

kdhoooon 2021. 5. 27. 22:45

문제


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

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

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

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

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

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

 

 

[제한 사항]

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

 

 

풀이


한방향으로 왼쪽으로만, 오른쪽으로만 스캔하여 최소크기를 return 했는데, 반례가있었다.

 

ABAAABA 의 경우

  • 한방향으로 스캔을 하면 7
  • 오른쪽 한번 이동 후 왼쪽 두번이동 하면, 5

5가 더 작은 경우이므로 5가 답이 돼야한다.

 

이 반례를 생각하지 못해서 정답이 나오지 않았다.

 

양방향으로 A가 아닌 알파벳이 있는 가장 가까운 곳으로 이동하는 식으로 코드를 작성해야한다.

이렇게 이동 후 숫자를 센 알파벳은 다시 찾아지지 않기 위해 'A' 로 바꿔주었다.

 

A 에서 위로 하나 올리면 Z 가 되는것도 고려하여  알파벳이 26개 이므로 nameChar[index] - 'A' 가 13보다 크면 26 - nameChar[index] - 'A' 를 하였다.

 

이 두가지 경우를 고려해서 문제를 풀면 된다.

 

<전체코드>

class Solution {
    
    static class Move{
        int count;
        int index;
        
        Move(int index, int count){
            this.count = count;
            this.index = index;
        }
    }
    
    public int solution(String name) {
        int answer = 0;
        
        char[] nameChar = name.toCharArray();
        
        Move move = new Move(0, 0);
        
        while(!isEnd(nameChar)){
            
            move = findMove(nameChar, move.index);
            
            if(nameChar[move.index] - 'A' > 13){
                answer += (26 - (nameChar[move.index] -'A')) + move.count;
            }
            else{
                answer += (nameChar[move.index] - 'A') + move.count;    
            }
            nameChar[move.index] = 'A';
        }
        
        return answer;
    }
    
    static boolean isEnd(char[] nameChar){
        
        for(int i = 0 ; i < nameChar.length ; i++){
            if(nameChar[i] != 'A')
                return false;
        }
        
        return true;
    }
    
    static Move findMove(char[] nameChar, int index){
        
        Move result;
        int min = 0;
        int minIndex = index;
        int i = index;
        int count = 0;
        
        while(true){
            if(nameChar[i] != 'A'){
                break;
            }
            i++;
            i %= nameChar.length;
            count++;
        }
        
        min = count;
        minIndex = i;
        
        i = index;
        count = 0;
        while(true){
            if(nameChar[i] != 'A'){ break; }
            
            i--;
            if(i < 0){
                i = nameChar.length - 1;
            }
            count++;
        }
        
        if(min > count){
            minIndex = i;
            min = count;
        }
        
        result = new Move(minIndex, min);
        
        return result;
    }
}