[프로그래머스] 대충 만든 자판 c++

문제 설명

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

예를 들어, 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]

풀이

keymap배열, targets배열의 요소를 하나하나 접급하여 비교하는 방법은 비효율적이고 시간복잡도를 초과할거 같아 다른 방법을 생각해보았습니다.

 

문제를 풀기위한 포인트는 "keymap 벡터의 어떤 요소이던지 먼저 나오는 알파벳의 순서로 키를 누른다." 입니다.

즉, keymap의 2차원 벡터에 저장되어 있는 요소들의 알파벳들을 index 빠른 순서로 저장하면 됩니다.

map 자료구조를 사용하여 (알파벳, 위치) 쌍으로 저장하도록 하였습니다.

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

using namespace std;

vector<int> solution(vector<string> keymap, vector<string> targets) {
    vector<int> answer;
    
    map<char, int> hash;
    for(const auto& it : keymap){
        for(int j = 0; j < it.size(); j++){
            if(hash[it[j]] > 0){
                hash[it[j]] = min(hash[it[j]] - 1, j) + 1;  //위치를 비교하기 위해 hash에 저장되어 있는 value 1감소
            }
            else{
                hash[it[j]] = j+1; //위치 + 1 = 버튼 클릭 횟수
            }
        }
    }
    
    for(const auto& it : targets){
        int cnt = 0;
        bool check = true;
        for(int j = 0; j < it.size(); j++){
            if(!check){
                continue;
            }
            if(hash[it[j]] < 1){
                check = false;
                cnt = -1;
            }
            else{
                cnt += hash[it[j]];

            }
        }
        answer.push_back(cnt);
    }
    
    return answer;
}