[프로그래머스 Level 1] 숫자 문자열과 영단어 (2021 카카오 기출)

문제

숫자 문자열과 영단어   

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

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

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

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

 

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

 

코드

#include <string>
#include <vector>

using namespace std;

int solution(string s) {
    int answer = 0;
    string alpha ="";
    
    vector<string> words = {"zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"};
    
    
    for(int i = 0; i < s.size(); i++)
    {
        if(s[i] >= '0' && s[i] <= '9')
            answer = answer * 10 + (s[i] - '0');
        else
        {
            alpha+=s[i];
            
            if(alpha.size() >= 3)
            {
                for(int j = 0; j < words.size(); j++)
                {
                    if(alpha.compare(words[j]) == 0)
                    {
                        answer = answer * 10 + j;
                        alpha.clear();
                    }    
                }
            }
                
        }
    }
    
    return answer;
}

 

나의 생각

정규 표현식으로 더 쉽게 풀 수 있는 사실을 알았다.

 

c++에서 제공하는 regex 라이브러리 사용법을 숙지하자.

 

멋진 코드

#include <string>
#include <vector>

using namespace std;

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

    string num_str[] = {"zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"};
    int len = s.length();

    for (int i = 0; i < len; i++) {
        if (s[i]  >= '0' && s[i] <= '9') {
            answer = answer * 10 + s[i] - '0';
        }
        else {
            for (int j = 0; j < 10; j++) {
                if(!num_str[j].compare(0, num_str[j].length(), s, i, num_str[j].length())) {
                    answer = answer * 10 + j;
                }
            }

        }
    }  

    return answer;
}

#include <string>
#include <vector>
#include <algorithm>
#include <regex>

using namespace std;

vector<string> numbers = {"zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine", "ten"};

int solution(string s) {
    int answer = 0;
    regex r;
    for(int i = 0; i < numbers.size(); i++) {
        r = numbers[i];
        s = regex_replace(s, r, to_string(i));
    }
    answer = stoi(s);
    return answer;
}