[Algorithm] 대충 만든 자판

문제 설명

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

 

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

 

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

 

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

 

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

 

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

 

문제 풀이

1. 각 알파벳의 입력 가능한 최소값을 구한다.
   1. [반복문] keymap을 순회한다.
      1. [반복문] keymap 요소인 문자열을 순회한다.
         1. 각 알파벳의 위치와 배열에 저장된 알파벳 위치를 비교하여 작은 값을 저장한다.
2. targets의 각 문자열의 문자를 몇 번 입력하여야 하는지 계산한다.
   
   1. 각 문자별로 배열에 저장된 최소 입력 값을 더해준다.
   2. 만약 입력할 문자가 없다면 -1을 결과값에 저장한다.
3. 결과값을 반환한다.

 

코드 작성

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

bool minCompare(int a, int b) {
    if (b < 0) return true;
    return a < b; 
}

int getAlphabetIdx(char ch)
{
    return ch - 'A';
}

vector<int> solution(vector<string> keymap, vector<string> targets) {
    vector<int> answer;
    vector<int> alphabetsDistance(26, -1);
    
    for (string str :keymap)
    {
        for (int idx = 0; idx < str.size(); idx++)
        {
            char alphabetIdx = getAlphabetIdx(str[idx]);
       alphabetsDistance[alphabetIdx] = min(alphabetsDistance[alphabetIdx], idx + 1, minCompare);
        }
    }
    

    for (string target : targets)
    {
        int targetTotal = 0;
        for (char ch : target)
        {
            char alphabetIdx = getAlphabetIdx(ch);
            if (alphabetsDistance[alphabetIdx] == -1)
            {
                targetTotal = -1;
                break;
            }
            else
            {
                targetTotal += alphabetsDistance[alphabetIdx];
            }
        }
        answer.push_back(targetTotal);
    }
    
    return answer;
}