[Java] 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
• [연산자]는 +와 -중 하나입니다.

 

class Solution {
    public String[] solution(String[] quiz) {
        String[] answer = new String[quiz.length];
        int idx = 0;
        
        for(String eval : quiz) {
            String[] quizArr = eval.split(" ");
            int result = Integer.parseInt(quizArr[0]);
            
            for(int i = 1; i < quizArr.length-2; i += 2) {
                if(quizArr[i].equals("+")) {
                    result += Integer.parseInt(quizArr[i+1]);
                }
                else if(quizArr[i].equals("-")) {
                    result -= Integer.parseInt(quizArr[i+1]);
                }
            }
            
            answer[idx] = (result == Integer.parseInt(quizArr[quizArr.length-1])) ? "O" : "X";
            idx++;
        }
        
        return answer;
    }
}

사실 수식의 길이가 X [연산자] Y = Z 로 정해져 있기 때문에 split을 한 후 0번 인덱스 [연산자] 1번 인덱스 = 결과비교 이런 식으로 해도 됐었다.

그래도 수식의 길이가 긴 경우를 포함해서 해결했다.

 

 

출처

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