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

풀이

지난번 풀었던 문자열 계산하기 문제를 응용했다. 주어진 'X [연산자] Y = Z' 형태의 수식이 QUIZ라는 배열에 하나 이상 존재할 수 있기 때문에 for문 안에서 split()으로 하나의 요소에 대한 배열을 새로 만들어 수식의 결과 여부를 판단했다.

그리고 공백을 기준으로 구분되어 만들어진 배열의 각 인덱스에는 숫자 혹은 연산자, 정해진 값이 들어 있다. 이에 0, 2, 4번 인덱스에 들어가 있는 숫자를 Integer.parseInt()로 파싱하여 변수에 할당하고, 1번 인덱스에 들어 있는 연산자가 -인지, +인지에 따라 분기한다. 마지막으로 값이 일치한지에 따라 삼항 연산자로 O, X를 나누어 문자열 배열 answer에 저장해 주었다.

import java.util.Arrays;

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

결과