[Java] 프로그래머스 LV.1 대충 만든 자판

영우·2026년 7월 18일

문제 설명

휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.

예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.

같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.

이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.

1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.

단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.

제한 사항

  • 1 ≤ keymap의 길이 ≤ 100
  • 1 ≤ keymap의 원소의 길이 ≤ 100
  • keymap[i]는 i + 1번 키를 눌렀을 때 순서대로 바뀌는 문자를 의미합니다.
  • 예를 들어 keymap[0] = "ABACD" 인 경우 1번 키를 한 번 누르면 A, 두 번 누르면 B, 세 번 누르면 A 가 됩니다.
  • keymap의 원소의 길이는 서로 다를 수 있습니다.
  • keymap의 원소는 알파벳 대문자로만 이루어져 있습니다.
  • 1 ≤ targets의 길이 ≤ 100
  • 1 ≤ targets의 원소의 길이 ≤ 100
  • targets의 원소는 알파벳 대문자로만 이루어져 있습니다.

입출력 예

KeymapTargetsResult
["ABACD", "BCEFD"]["ABCD", "AABB"][9, 4]
["AA"]["B"][-1]
["AGZ", "BSSS"]["ASA", "BGZ"][4, 6]

풀이

import java.util.*;
class Solution {
    public int[] solution(String[] keymap, String[] targets) {
        int[] answer = new int[targets.length];
        
        Map<Character, Integer> map = new HashMap<>();
        for(String key : keymap) {
            for(int i = 0; i < key.length(); i++) {
                char c = key.charAt(i);
                
                if(map.containsKey(c)) {
                    map.put(c, Math.min(map.get(c), i + 1));
                } else {
                
                    map.put(c, i + 1);
                }
            }
        }
        
        int idx = 0;
        for(String target : targets) {
            int sum = 0;
            for(int i = 0; i < target.length(); i++) {
                char c = target.charAt(i);
                
                if(map.containsKey(c)) {
                    sum += map.get(c);
                } else {
                    sum = -1;
                    break;
                }
            }
            
            answer[idx++] = sum;
        }
        
        return answer;
    }
}

처음에는 각 target의 문자마다 모든 keymap을 탐색해서 최소 입력 횟수를 찾으려고 생각했다. 하지만 이렇게 하면 비효율적이다.

keymap = ["ABACD", "BCEFD"]

target = "ABCD"

A의 최소 입력 횟수는 항상 1이고, B는 항상 1이다.
즉 한 번 구해놓으면 계속 재사용할 수 있었다.

그래서 먼저 각 문자의 최소 입력 횟수를 저장하는 Map을 만든 뒤 target을 순회하면서 값을 더해주는 방식으로 해결했다.

if(map.containsKey(c)) {
    map.put(c, Math.min(map.get(c), i + 1));
} else {
    map.put(c, i + 1);
}

keymap을 순회하면서 이미 존재하는 문자라면 Math.min(기존값, 현재 위치 + 1) 으로 더 작은 값을 유지했다.

for(char c : target.toCharArray()) {
    if(map.containsKey(c)) {
        sum += map.get(c);
    } else {
        sum = -1;
        break;
    }
}

target 문자열을 순회하면서 Map에 존재하면 입력 횟수를 더해주었고 존재 하지 않으면 -1를 반환하였다.

0개의 댓글