[c++] 프로그래머스 - 숫자 문자열과 영단어( Level 1, 2021 카카오 인턴십)

문제

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

 

코드

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

char mapping(string number){
    if(number =="zero") return '0';
    if(number =="one") return '1';
    if(number =="two") return '2';
    if(number =="three") return '3';
    if(number =="four") return '4';
    if(number =="five") return '5';
    if(number =="six") return '6';
    if(number =="seven") return '7';
    if(number =="eight") return '8';
    if(number =="nine") return '9';
    return '0';
}

bool check(string number){
    for(int i=0; i<11; i++){
        if(number =="zero") return true;
        if(number =="one") return true;
        if(number =="two") return true;
        if(number =="three") return true;
        if(number =="four") return true;
        if(number =="five") return true;
        if(number =="six") return true;
        if(number =="seven") return true;
        if(number =="eight") return true;
        if(number =="nine") return  true;
    }
    return false;
}
int solution(string s) {
    int answer = 0;
    string temp = "", number ="";
    
    for(int i=0; i<s.length()+1; i++){
        bool flag = check(temp);
        char mapping_num;
        if(flag){
            number+=mapping(temp);
            temp.clear();
        }
        if(s[i]>='a'&&s[i]<='z') temp+=s[i];
        else{
            if(!temp.empty()){
                number +=mapping_num;
            }
            number+=s[i];
            temp.clear();
        }
    }
    answer = stoi(number);
    return answer;

 

풀이(19분)

문제 자체는 쉬웠지만 코드가 너무 지저분했습니다. 그래서 다른 사람의 풀이를 보다가 c++의 regex 정규식 메서드를 알게 되었고, 아래와 같이 코드를 작성할 수 있었습니다.

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

using namespace std;

int solution(string s) {
    int answer = 0;
    
    s = regex_replace(s, regex("zero"),"0");
    s = regex_replace(s, regex("one"),"1");
    s = regex_replace(s, regex("two"),"2");
    s = regex_replace(s, regex("three"),"3");
    s = regex_replace(s, regex("four"),"4");
    s = regex_replace(s, regex("five"),"5");
    s = regex_replace(s, regex("six"),"6");
    s = regex_replace(s, regex("seven"),"7");
    s = regex_replace(s, regex("eight"),"8");
    s = regex_replace(s, regex("nine"),"9");
    answer = stoi(s);
    return answer;
}

코드가 훨씬 간결해졌습니다.