[프로그래머스] OX퀴즈 (C++)

OX퀴즈 / Lv.0

---

문제  설명 )

덧셈, 뺄셈 수식들이 '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
• [연산자]는 + 와 - 중 하나입니다.

 

 

 

입출력 예 )

 

 

입출력 예 설명 )

입출력 예 #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합니다.

 

 

풀이)

 

새벽에 좀 비몽사몽 풀었더니 코드가 깔끔하지가 않다..

 

#include <string>
#include <vector>
 
using namespace std;
 
vector<string> solution(vector<string> quiz) {
    vector<string> answer;
 
    // 문자열 배열 순회
    for(int i = 0; i < quiz.size(); i++)
    {
        // 연산자 
        string opt;
        // 임시 string
        string temp = "";
        // 나오는 숫자를 담을 vec
        vector<int> vec;
        
        // 해당 문자열 순회
        for(int j = 0; j < quiz[i].length(); j++)
        {
            // 문자열이 끝났다면 마지막 temp에 들어있는것은 숫자이므로
            // vec에 넣어준다.
            if (j == quiz[i].length() - 1)
            {
                temp += quiz[i][j];
                vec.push_back(stoi(temp));
                continue;
            }
            
            // 공백을 만났을 때.
            if(quiz[i][j] == ' ')
            {
                // temp가 연산자라면 opt를 해당 연산자로 바꿔주고,
                // temp 초기화
                if(temp == "-") 
                {
                    opt = "-";
                    temp = "";
                    continue;
                }
                
                if(temp == "+") 
                {
                    opt = "+";
                    temp = "";
                    continue;
                }
                
                if(temp == "=")
                {
                    temp = "";
                    continue;
                }
                
                // 연산자가 아니라면 temp는 숫자이므로 vec에 넣는다.
                vec.push_back(stoi(temp));
                temp = "";
                continue;
            }
            
            // 공백이 아닐 경우 temp를 늘려감
            temp += quiz[i][j];
        }
        
        // opt에 따라서 해당 수식이 맞는지를 판단함.
        if (opt == "-")
        {
            if(vec[0] - vec[1] == vec[2])
            {
                answer.push_back("O");
            }
            else answer.push_back("X");
        }
        else if (opt == "+")
        {
            if(vec[0] + vec[1] == vec[2])
            {
                answer.push_back("O");
            }
            else answer.push_back("X");
        }
    }
    
    return answer;
}

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120907

프로그래머스