C++ / 프로그래머스 / [3차] 압축

문제

 

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

프로그래머스

문제풀이

 

사전에 1: A부터 26: Z까지 이미 있으므로 벡터를 만들어 알파벳을 저장하고, 반복문을 이용해 unordered_map에 저장한다.

 

string인 msg를 탐색하면서 맨 앞부터 한 자리씩 string변수 str에 더한다.

str이 사전(unordred_map)에 있다면 prev에 str을 저장한다.

만약 str이 사전에 없다면 prev까지는 사전에 있으므로 사전에서 prev값을 찾아 answer에 push 한다.

그리고 사전에 str값을 사전의 크기+1로 입력한다.

 

코드

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

using namespace std;

vector<int> solution(string msg) {
    vector<int> answer;
    vector<string> alp = { "A","B","C","D","E","F","G","H","I","J","K","L","M","N","O","P","Q","R","S","T","U","V","W","X","Y","Z" };

    unordered_map<string, int>dic;
    for (int idx = 0; idx < 26; ++idx) {
        dic[alp[idx]] = idx + 1;
    }
    msg += " ";
    string str("");
    string prev("");
    for (int idx = 0; idx < msg.size(); ++idx) {
        str += msg[idx];
        if (dic.find(str) == dic.end()) {
            answer.push_back(dic[prev]);
            dic[str] = dic.size() + 1;
            prev = msg[idx];
            str = msg[idx];
        }
        else prev = str;
    }

    return answer;
}