[프로그래머스][해시][C++] 숫자 문자열과 영단어

숫자 문자열과 영단어

https://programmers.co.kr/learn/courses/30/lessons/81301

코딩테스트 연습 - 숫자 문자열과 영단어

---

문제 설명

네오와 프로도가 숫자놀이를 하고 있습니다. 네오가 프로도에게 숫자를 건넬 때 일부 자릿수를 영단어로 바꾼 카드를 건네주면 프로도는 원래 숫자를 찾는 게임입니다.

다음은 숫자의 일부 자릿수를 영단어로 바꾸는 예시입니다.

• 1478 → "one4seveneight"
• 234567 → "23four5six7"
• 10203 → "1zerotwozero3"

이렇게 숫자의 일부 자릿수가 영단어로 바뀌어졌거나, 혹은 바뀌지 않고 그대로인 문자열 s가 매개변수로 주어집니다. s가 의미하는 원래 숫자를 return 하도록 solution 함수를 완성해주세요.

참고로 각 숫자에 대응되는 영단어는 다음 표와 같습니다.

숫자	영단어
0	zero
1	one
2	two
3	three
4	four
5	five
6	six
7	seven
8	eight
9	nine

제한 사항

• 1 ≤ s의 길이 ≤ 50
• s가 "zero" 또는 "0"으로 시작하는 경우는 주어지지 않습니다.
• return 값이 1 이상 2,000,000,000 이하의 정수가 되는 올바른 입력만 s로 주어집니다.

풀이 방법

문자열 직접 비교로 해결할 수도 있겠지만 나는 해시맵을 사용하는게 더 편할 것 같다고 생각했다.(반복적인 find 연산으로 시간복잡도는 떨어질 수도 있겠다)

unordered_map<string, int>를 선언해 영단어와 숫자를 Key, Value 값으로 넣어주고 문자열을 탐색하면서 맵에서 find한 영단어를 Value값으로 바꿔주었다.

소스 코드

#include <string>
#include <vector>
#include <unordered_map>
#include <math.h>

using namespace std;

unordered_map<string, int> mapInit() {
    unordered_map<string, int> um;

    um["zero"] = 0;
    um["one"] = 1;
    um["two"] = 2;
    um["three"] = 3;
    um["four"] = 4;
    um["five"] = 5;
    um["six"] = 6;
    um["seven"] = 7;
    um["eight"] = 8;
    um["nine"] = 9;

    return um;
}

int solution(string s) {
    int answer = 0;
    string tmp = "";

    unordered_map<string, int> nm;

    nm = mapInit();

    for(int i = 0; i < s.size(); i++) {
        tmp = "";

        if(s[i] > 9)
            for(int j = i; j < s.size(); j++) {
                if(s[j] <= 9)
                    break;
                tmp += s[j];
                auto iter = nm.find(tmp);
                if(iter != nm.end()) {
                    s.erase(i + 1, j - i);
                    s[i] = iter->second + '0';
                }
            }
    }

    for(int i = 0; i < s.size(); i++)
        answer += (s[i] - '0') * pow(10, s.size() - i - 1);

    return answer;
}