프로그래머스 : 대충만든자판(56/Lv3)

문제) 대충만든자판

https://school.programmers.co.kr/learn/courses/30/lessons/160586

 

keymap에 자판 규칙이, targets에 쳐야하는 String값들이 들어온다.

이것을 최소한을 쳐서 몇 번만에 targets을 칠 수 있는지 반환한다 (칠 수 없는 경우 -1반환)

 

나의 경우 해당 문제를 보자마자 배열에 A~Z까지 최소 몇 번 만에 도달하는지 저장 한 뒤 alphabetCount배열에 미리 저장한뒤,

target에서 그 배열값을 보며 접근하는 것이 빠르겠다고 생각하여 그렇게 코드를 짜게 되었다.

그 결과 성공적으로 아래와 같이 코드를 짤 수 있었다. 

class Solution {
    public int[] solution(String[] keymap, String[] targets) {
        int[] answer = new int[targets.length];
        int[] alphabetCount = new int[26];
        
        // keymap에 대한 alphabetCount 배열을 미리 생성하여 
        // 각 알파벳 마다 최소의 경우로 도달하는 수를 저장해 놓는다. 
        for(int k = 0; k < keymap.length; k++){
            String s = keymap[k];
            for(int i = 0; i < s.length(); i++){
                int location = (int) s.charAt(i) - 65;
                if(alphabetCount[location] == 0 
                   || alphabetCount[location] > i + 1){
                    alphabetCount[location] = i + 1;
                }
            }
        }
        
        // targets배열을 이중for문을 돌려 alphabetCount에서 바로 값을 꺼내와 sum에 다 더한다.
        for(int i=0; i<targets.length; i++){
            int sum = 0;
            for(int j=0; j<targets[i].length(); j++){
                // 0인 경우는 알파벳이 keymap에 없는 경우이므로 아래와 같이 처리한다. 
                if(alphabetCount[(int) targets[i].charAt(j) - 65] == 0){
                    sum = -1; 
                    break;
                }
                sum += alphabetCount[(int) targets[i].charAt(j) - 65];
            }
            answer[i] = sum;
        }
        
        return answer;
    }
}