[프로그래머스/C++ 문제 풀이] Lv. 0 - OX퀴즈

문제 설명

덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 "O"를 틀리다면 "X"를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한 사항

• 연산 기호와 숫자 사이는 항상 하나의 공백이 존재합니다. 단 음수를 표시하는 마이너스 기호와 숫자 사이에는 공백이 존재하지 않습니다.
• 1 ≤ quiz의 길이 ≤ 10
• X, Y, Z는 각각 0부터 9까지 숫자로 이루어진 정수를 의미하며, 각 숫자의 맨 앞에 마이너스 기호가 하나 있을 수 있고 이는 음수를 의미합니다.
• X, Y, Z는 0을 제외하고는 0으로 시작하지 않습니다.
• -10,000 ≤ X, Y ≤ 10,000
• -20,000 ≤ Z ≤ 20,000
• [연산자]는 + 와 - 중 하나입니다.

입출력 예

quiz	result
["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"]	["O", "O", "X", "O"]
["3 - 4 = -3", "5 + 6 = 11"]	["X", "O"]

 

입출력 예 #1

• 3 - 4 = -3 은 틀린 수식이므로 "X", 5 + 6 = 11 은 옳은 수식이므로 "O" 입니다. 따라서 ["X", "O"]를 return합니다.

입출력 예 #2

• 19 - 6 = 13 은 옳은 수식이므로 "O", 5 + 66 = 71 은 옳은 수식이므로 "O", 5 - 15 = 63 은 틀린 수식이므로 "X", 3 - 1 = 2는 옳은 수식이므로 "O" 따라서 ["O", "O", "X", "O"]를 return합니다.

---

문제 풀이

이번 문제는 문자열에서 특정 정보를 추출하여 결과물을 얻을 수 있는지를 판별하기 위한 문제입니다.

저는 string 클래스에서 제공하는 substr(), find(), erase() 함수를 통해 문제를 해결하였습니다.

#include <string>
#include <vector>

using namespace std;

vector<string> solution(vector<string> quiz) {
    vector<string> answer;

    for (string str : quiz)
    {
        string num1 = str.substr(0, str.find(" "));
        str.erase(0, str.find(" ") + 1);

        string op = str.substr(0, str.find(" "));
        str.erase(0, str.find(" ") + 1);

        string num2 = str.substr(0, str.find(" "));
        str.erase(0, str.find("=") + 2);

        if (op == "+")
        {
            if (to_string(stoi(num1) + stoi(num2)) == str)
            {
                answer.push_back("O");
            }
            else
            {
                answer.push_back("X");
            }
        }
        else if (op == "-")
        {
            if (to_string(stoi(num1) - stoi(num2)) == str)
            {
                answer.push_back("O");
            }
            else
            {
                answer.push_back("X");
            }
        }
    }

    return answer;
}

 

하지만.. 이후 찾아보니 문자열에서 필요한 자료형에 맞는 정보를 꺼낼 때 유용하게 사용되는 stringstream 클래스가 있었으며, 더욱 간편하게 작업이 가능하다는 사실을 알게 되었습니다.

 

stringstream 클래스 간단 설명

stringstream는 C++에서 문자열 스트림을 다루기 위한 클래스로, <sstream> 헤더 파일에 정의되어 있습니다.

위 문제에서 해당 클래스는 다음과 같이 숫자 및 연산자를 추출하기 위해 사용될 수 있으며, 추출 시 각 문자열은 공백을 기준으로 나누어지게 됩니다.

// stringstream 객체 생성
stringstream ss;

// 사용할 문자열 설정
ss.str(s);

// 문자열 파싱을 위한 변수 선언
int num1;
int num2;
int result;
char oper;
char equl;

// 값 읽어와서 저장하기 (공백 및 줄바꿈 문자열은 무시)
ss >> num1 >> oper >> num2 >> equl >> result;

 

이후 operator의 값에 따른 연산을 통해 result의 값과 같은지 체크하고 결과를 반환하면 쉽고 간편하게 문제를 해결할 수 있습니다.