[연습문제] 대충 만든 자판

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

프로그래머스

---

문제 설명
휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다.

키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.

• 예를 들어, 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의 원소는 알파벳 대문자로만 이루어져 있습니다.

---

입출력 예

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

---

입출력 예 설명
입출력 예 #1

 

"ABCD"의 경우,

• 1번 키 한 번 → A
• 2번 키 한 번 → B
• 2번 키 두 번 → C
• 1번 키 다섯 번 → D

따라서 총합인 9를 첫 번째 인덱스에 저장합니다.

 

"AABB"의 경우,

• 1번 키 한 번 → A
• 1번 키 한 번 → A
• 2번 키 한 번 → B
• 2번 키 한 번 → B

따라서 총합인 4를 두 번째 인덱스에 저장합니다.

결과적으로 [9,4]를 return 합니다.

 

입출력 예 #2

"B"의 경우, 'B'가 어디에도 존재하지 않기 때문에 -1을 첫 번째 인덱스에 저장합니다.
결과적으로 [-1]을 return 합니다.

 

입출력 예 #3

 

"ASA"의 경우,

• 1번 키 한 번 → A
• 2번 키 두 번 → S
• 1번 키 한 번 → A

따라서 총합인 4를 첫 번째 인덱스에 저장합니다.

 

"BGZ"의 경우,

• 2번 키 한 번 → B
• 1번 키 두 번 → G
• 1번 키 세 번 → Z

따라서 총합인 6을 두 번째 인덱스에 저장합니다.

결과적으로 [4, 6]을 return 합니다.

---

문자열들이 담긴 배열 targets가 주어졌을 때, 키패드 배열 keymaps에 있는 알파벳들을 활용해서 문자열을 만들 수 있는 최소 푸시 횟수를 구하는 문제입니다.

 

처음 이 문제를 보았을 때, 키패드에 있는 문자들은 정해져 있기 때문에 index나 find를 사용해서 구하면 되지 않을까 하는 생각이 들었습니다.

그리고 최소 횟수라는 점에서 key - value의 해시 테이블, 혹은 딕셔너리 구조를 사용하면 항상 최소값을 보장할 수 있을 거라 생각했습니다.

자바에서는 해시맵을, 파이썬에서는 딕셔너리를 사용해서 문제를 풀어보겠습니다.

 

1. 자바

import java.util.HashMap; // 해시맵을 위한 패키지 임포트

class Solution {
    public int[] solution(String[] keymap, String[] targets) {
        int[] answer = new int[targets.length];
        // 해시맵은 래퍼 클래스를 사용합니다.
        HashMap<Character, Integer> hmap = new HashMap<>();

        // keymap을 순회하여 최소 푸시 횟수를 기록
        for (String s : keymap) {
            for (char c : s.toCharArray()) {
                // 키가 이미 존재하면 최소값을 갱신
                if (hmap.containsKey(c)) {
                    hmap.put(c, Math.min(hmap.get(c), s.indexOf(c) + 1));
                } else {
                    // 처음 등장하는 키는 초기값 설정
                    hmap.put(c, s.indexOf(c) + 1);
                }
            }
        }

        // targets를 순회하며 결과 계산
        for (int i = 0; i < targets.length; i++) {
            int total_presses = 0;
            for (char c : targets[i].toCharArray()) {
                Integer value = hmap.get(c);
                if (value == null) {
                    total_presses = -1;
                    break; // 작성 불가능한 경우 즉시 종료
                }
                total_presses += value;
            }
            answer[i] = total_presses;
        }

        return answer;
    }
}

 

2. 파이썬

from collections import defaultdict
def solution(keymap, targets):
    answer = []
    # 각 알파벳을 누르기 위한 최소 횟수 (기본값: 101)
    minpush = defaultdict(lambda: 101)
    # 모든 키패드에 대해 탐색합니다.
    for key in keymap:
        # 해당 키패드의 알파벳을 하나씩 찾습니다.
        for i, char in enumerate(key):
            # 알파벳의 최소 푸시 횟수를 기록합니다.
            minpush[char] = min(minpush[char], i + 1)
            
    # 문자열에 대해 탐색을 시작합니다.
    for target in targets:
        total_presses = 0 # 각 문자열을 만들기 위한 푸시 횟수
        for char in target: # 알파벳별 탐색
            if minpush[char] == 101: # 101 (조건상 불가능)
                total_presses = -1 # 생성 불가 -1
                break
            total_presses += minpush[char] # 알파벳에 추가합니다.
        answer.append(total_presses) # 최소 횟수를 정답에 기재
        
    return answer

 

문자열을 만들기 위해서는 문자열의 알파벳들이 모두 키패드에 존재해야 합니다.

따라서 키패드를 통해서 먼저 사용할 수 있는 알파벳들의 최소 푸시 횟수를 해시맵 or 딕셔너리 구조로 저장한 뒤, 문자열의 알파벳을 순회하면서 해당 값이 있는지 없는지 확인하면 됩니다.

 

자바에서는 해시맵의 get을 사용해서, 파이썬에서는 defaultdict의 초기값 설정을 통해서 가능/불가능 여부를 확인했습니다.