문제 링크 https://programmers.co.kr/learn/courses/30/lessons/81301

문제

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

  • 1478 → “one4seveneight”
  • 234567 → “23four5six7”
  • 10203 → “1zerotwozero3”

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

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

제한사항

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

풀이 과정

일단 string형의 빈 str을 준비한다.
그리고 substr을 사용하여 각 영단어에 대응하는 숫자를 문자열에 더해줬다.

마지막에 int로 반환을 해야되니 stoi함수를 써서 string을 int형으로 바꿔주었다.

#include <string>
#include <vector>

using namespace std;

int solution(string s) {
    int answer = 0;
    string str = "";
    for(int i = 0; i < s.size(); i++){
        if(s.substr(i, 4) == "zero"){
            str += "0";
        }
        else if(s.substr(i, 3) == "one"){
            str += "1";
        }
        else if(s.substr(i, 3) == "two"){
            str += "2";
        }
        else if(s.substr(i, 5) == "three"){
            str += "3";
        }
        else if(s.substr(i, 4) == "four"){
            str += "4";
        }
        else if(s.substr(i, 4) == "five"){
            str += "5";
        }
        else if(s.substr(i, 3) == "six"){
            str += "6";
        }
        else if(s.substr(i, 5) == "seven"){
            str += "7";
        }
        else if(s.substr(i, 5) == "eight"){
            str += "8";
        }
        else if(s.substr(i, 4) == "nine"){
            str += "9";
        }
        else if("0" <= s.substr(i, 1) && s.substr(i, 1) <= "9"){
            str += s[i];
        }
    }
    answer = stoi(str);
    return answer;
}

댓글남기기