C++ / 프로그래머스 / 대충 만든 자판

문제

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

프로그래머스

문제풀이

해당 문자를 작성하기 위한 최소 횟수를 구하기 위해, 알파벳 개수가 크기인 cnt벡터를 선언한다.

해당 문자가 없을 때는 -1을 출력해야 하기 때문에, 값들은 최대 횟수+1인 101로 모두 저장한다.

알파벳은 대문자로만 이루어져 있으므로, keymap을 탐색하면서 해당 문자열 - 'A'한 값과 기존 cnt값 중 작은 값을 cnt에 저장한다.

targets을 탐색하면서 해당 문자열 횟수를 cnt에서 구하고 값을 sum에 계속 저장한다.

만약 자판에 해당 문자가 없다면 값은 101이 저장되어 있으므로 -1을 반복문을 종료하고, answer에 저장한다.

 

코드

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> solution(vector<string> keymap, vector<string> targets) {
    int y(0), x(0), sum(0);
    vector<int> answer;
    vector<int> cnt(26, 101);

    for (y = 0; y < keymap.size(); ++y) {
        for (x = 0; x < keymap[y].size(); ++x) {
            cnt[keymap[y][x] - 'A'] = min(cnt[keymap[y][x] - 'A'], x + 1);
        }
    }
    for (y = 0; y < targets.size(); ++y) {
        sum = 0;
        for (x = 0; x < targets[y].size(); ++x) {
            if (cnt[targets[y][x] - 'A'] == 101) {
                sum = -1;
                break;
            }
            sum += cnt[targets[y][x] - 'A'];
        }
        answer.push_back(sum);
    }
    return answer;
}