프로그래머스 Lv1 대충 만든 자판

문제

https://school.programmers.co.kr/learn/courses/30/lessons/160586?language=cpp

휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다. 예를 들어, 1번 키에 “A”, “B”, “C” 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 “A”, 두 번 누르면 “B”, 세 번 누르면 “C”가 되는 식입니다. 같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다. 이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다. 1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요. 단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.

수도 코드

1. 문자별 최소 누름 횟수를 저장할 빈 해시맵(char_min_press)을 만든다.

2. 모든 keymap을 순회하면서,
   각 키(key)에 대해 다음을 수행한다
    - key에 있는 문자를 한 글자씩 확인하면서
    - 해당 문자가 처음 나온 경우, i + 1 (i는 인덱스 → 누름 횟수)을 저장한다.
    - 이미 저장된 문자가 다시 나오면, 기존 값과 비교해서 더 작은 값으로 갱신한다.
      (즉, 해당 문자를 누르기 위한 최소 누름 횟수를 저장)

3. 각 target 문자열을 순회하면서
    - 각 문자열마다 누름 수 총합(total)을 0으로 초기화하고,
    - 문자열을 구성하는 모든 문자를 하나씩 확인한다
        - 해당 문자가 char_min_press에 있다면
            - 저장된 누름 횟수를 total에 누적한다.
        - 없다면:
            - 해당 문자열은 입력할 수 없으므로 can_type = false로 설정하고, 중단한다.
    - 문자열을 끝까지 확인한 뒤,
        - can_type이 true면 total을 answer에 추가하고,
        - false면 -1을 추가한다.

4. answer를 반환한다.

코드

#include<bits/stdc++.h>
using namespace std;
unordered_map<char,int> char_min_press;

vector<int> solution(vector<string> keymap, vector<string> targets) {
    vector<int> answer;
    for(auto key : keymap){
        for(int i = 0; i < key.size(); i++){
            char c = key[i];
            int press_count = i + 1;
            if(char_min_press.count(c)){
                char_min_press[c] = min(char_min_press[c], press_count);
            } else {
                char_min_press[c] = press_count;
            }
        }
    }

    for(auto a : targets){
        int total = 0;
        bool can_type = true;
        for(auto c : a){
            if(char_min_press.count(c)){
                total += char_min_press[c];
            } else {
                can_type = false;
                break;
            }
        }
        answer.push_back(can_type ? total : -1);
    }

    return answer;
}

Reference

나의 뇌