프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

📄 문제

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

 

📝 풀이

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

vector<int> solution(vector<string> keymap, vector<string> targets) {
    
    unordered_map<char, int> m;
    for(string s : keymap)
        for(int i = 0; i < s.size(); ++i)
        {
            // 이미 값이 존재하는경우
            if(m.find(s[i]) != m.end())
                m[s[i]] = min(m[s[i]], i + 1);
            else
                m[s[i]] = i + 1;
        }
            
    vector<int> answer(targets.size(), 0);
    for(int i = 0; i < targets.size(); ++i)
        for(int j = 0; j < targets[i].size(); ++j)
        {
            if(m.find(targets[i][j]) != m.end())
                answer[i] += m[targets[i][j]];
            else
            {
                answer[i] = -1;
                break;
            }
        }
            
    return answer;
}
  • map을 이용하여 keymap에서 중복되는 모든 문자에 대한 최소 클릭 횟수를 갱신하여 저장합니다.
  • targets를 반복문에서 각 문자열에 대해 keymap에서 value를 획득하여 값을 더하거나, keymap에 해당 키가 없다면, -1를 저장하도록 합니다.
bonnate