[코딩 테스트] 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
["3 - 4 = -3", "5 + 6 = 11"], ["X", "O"]
["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"], ["O", "O", "X", "O"]
풀이
#include <string>
#include <vector>
#include <sstream>
vector<string> solution(vector<string> quiz) {
vector<string> answer(quiz.size());
int num1, num2, num3;
char calcu, garbage;
for (int i = 0; i < quiz.size(); i++){
stringstream ss(quiz[i]);
ss >> num1 >> calcu >> num2 >> garbage >> num3;
if (calcu == '+') answer[i] = ((num1+num2) == num3) ? "O" : "X";
else answer[i] = ((num1-num2) == num3) ? "O" : "X";
ss.clear();
}
return answer;
}
answer 벡터를 quiz 벡터의 크기와 같게 모두 0으로 초기화한다.
문제에서 quiz 벡터를 공백 기준으로 나누면 총 5개의 원소가 되는데, 이때 각 원소는 제한 조건에 따라
순서에 맞게 타입이 정해져 있다. 따라서 sstream으로 타입에 맞게 배분해준다.
추가로 설명할 만한 부분은 없지만, 반환 타입에 주의하여 char과 string을 잘 표시하여야 한다. ( ‘c’, “string” )
문제 제한 사항 이용