[PS] 99클럽 코테 스터디 24일차 TIL (대충 만든 자판)

문제

문제 링크

설명

키패드 배열이 keymap 으로 주어지고, 입력하고자 하는 값이 targets 배열로 주어질 때 targets 에 들어있는 단어를 완성하기 위해 키패드를 총 몇번씩 눌러야 하는지 각각 누적 횟수를 반환하는 문제

풀이

키패드를 최소로 눌러야하기 때문에 주어진 keymap 전체에서 중복되는 문자는 인덱스 번호가 더 작은 키패드를 눌러야 한다.

keymap = ["ABACD", "BCEFD"]
targets = ["ABCD"]

예를 들어 keymap 과 targets 배열이 위와 같을때 A 가 키패드 하나에 두개 들어있지만 처음 한번만 누르면 된다. 각 키패드의 순서를 저장할 map 을 만들어 두고, keymap 의 모든 항목을 순회하면서 중복 문자에 대해선 순서가 더 작은 값으로 갱신시켜 준다.

이때 순서는 1부터 저장한다. 이제 문자 별 눌러야 하는 최소 횟수가 map 에 모두 들어있게 된다. map 에서 target 에 있는 문자를 찾아서 카운트만 해주면 된다.

class Solution {
    public int[] solution(String[] keymap, String[] targets) {
        int[] answer = new int[targets.length];
        HashMap<Character, Integer> map = new HashMap<>();
        for (String s : keymap) {
          char[] charArray = s.toCharArray();
          for (int j = 0; j < charArray.length; j++) {
            char c = charArray[j];
            if (map.get(c) == null || map.get(c) > j) {
              map.put(c, j + 1);
            }
          }
        }

        for (int i = 0; i < targets.length; i++) {
          char[] charArray = targets[i].toCharArray();
          for (int j = 0; j < charArray.length; j++) {
            char c = charArray[j];
            if (map.containsKey(c)) {
              answer[i] += map.get(c);
            } else {
              answer[i] = -1;
              break;
            }
          }
        }
        return answer;
    }
}