[프로그래머스] 2018 KAKAO BLIND RECRUITMENT - [3차] 압축, C++

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

코딩테스트 연습 - [3차] 압축 | 프로그래머스

 본 문제는 '2018 KAKAO BLIND RECRUITMENT'로 kakao Tech에 의하면 "정답률 95.80%, 가장 많은 지원자가 잘 품. 언어별로는 Java 언어 사용자들이 조금 어려워함." 라고 한다. map을 이용하여 문제를 풀었고, 제시된 조건만 잘 구현하면 되는 문제다.

#include <string>
#include <vector>
#include <map>
#include <iostream>
 
using namespace std;
 
vector<int> solution(string msg) {
    vector<int> answer;
    int index = 0;
    map<string, int> m;
    // 길이가 1인 모든 단어를 포함하도록 초기화.
    m.insert(make_pair("A", 1));m.insert(make_pair("B", 2));
    m.insert(make_pair("C", 3));m.insert(make_pair("D", 4));
    m.insert(make_pair("E", 5));m.insert(make_pair("F", 6));
    m.insert(make_pair("G", 7));m.insert(make_pair("H", 8));
    m.insert(make_pair("I", 9));m.insert(make_pair("J", 10));
    m.insert(make_pair("K", 11));m.insert(make_pair("L", 12));
    m.insert(make_pair("M", 13));m.insert(make_pair("N", 14));
    m.insert(make_pair("O", 15));m.insert(make_pair("P", 16));
    m.insert(make_pair("Q", 17));m.insert(make_pair("R", 18));
    m.insert(make_pair("S", 19));m.insert(make_pair("T", 20));
    m.insert(make_pair("U", 21));m.insert(make_pair("V", 22));
    m.insert(make_pair("W", 23));m.insert(make_pair("X", 24));
    m.insert(make_pair("Y", 25));m.insert(make_pair("Z", 26));
    
    int idx = 27;
    // 가장 긴 문자열 찾기
    while (index != msg.size()) {
        string chk = "";
        char now = msg[index];
        chk += now;
        char next = msg[index+1];
        // 다음 글자(c)가 포함된 문자열이 없을 때
        if (m.find(chk+next) == m.end()) {
            answer.push_back(m.find(chk)->second);
            m.insert(make_pair(chk+next, idx));
            idx += 1;
        }
        // 다음 글자(c)가 포함된 문자열이 있을 때, 다음 글자(c)를 추가
        else {
            while (m.find(chk+next) != m.end()) {
                chk += next;
                index += 1;
                next = msg[index+1];
            }
            answer.push_back(m.find(chk)->second);
            m.insert(make_pair(chk+next, idx));
            idx += 1;
        }
        index += 1;
    }
 
    return answer;
}