[프로그래머스/C++] 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
• [연산자]는 + 와 - 중 하나입니다.

---

입출력 예quizresult

["3 - 4 = -3", "5 + 6 = 11"]	["X", "O"]
["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"]	["O", "O", "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합니다.

---

우선 for문 내에서 공백 기준으로 문자열을 나눠서 새로운 벡터에 넣는다.

 

        istringstream ss(quiz[i]);
        string stringBuffer;
        vector<string> x;
        x.clear();
        while (getline(ss, stringBuffer, ' ')){
            x.push_back(stringBuffer);
        }

 

첫번째 예시의 quiz[0]을 예로들면

x 벡터에 ["3", "-", "4", "=", "-3"]

이렇게 들어가 있을 것임.

수식의 형태는 항상 동일하니,

숫자와 연산자의 위차가 항상 동일하다는 것을 생각하면 쉽게 풀수 있음.

 

#include <string>
#include <vector>
#include <sstream>

using namespace std;

vector<string> solution(vector<string> quiz) {
    vector<string> answer;
    
    
    for(int i = 0; i < quiz.size(); i++)
    {
        istringstream ss(quiz[i]);
        string stringBuffer;
        vector<string> x;
        x.clear();
        while (getline(ss, stringBuffer, ' ')){
            x.push_back(stringBuffer);
        }

        int tmp = 0;
        
        if(x[1] == "-")
        {
            tmp = stoi(x[0]) - stoi(x[2]);
        }
        
        else if(x[1] == "+")
        {
            tmp = stoi(x[0]) + stoi(x[2]);
        }
        
        if(tmp == stoi(x[4]))
        {
            answer.push_back("O");
        }
        else
        {
            answer.push_back("X");
        }
            
    }
    
    return answer;
}