OX 퀴즈

/* OX 퀴즈

 * 덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다.

 * 수식이 옳다면 "O"를 틀리다면 "X"를

 * 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요

 *

 * 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"]

 *

 * 3 - 4 = -3 은 틀린 수식이므로 "X", 5 + 6 = 11 은 옳은 수식이므로 "O" 입니다.

 * 따라서 ["X", "O"]를 return합니다.

 * 19 - 6 = 13 은 옳은 수식이므로 "O", 5 + 66 = 71 은 옳은 수식이므로 "O",

 * 5 - 15 = 63 은 틀린 수식이므로 "X", 3 - 1 = 2는 옳은 수식이므로 "O"

 * 따라서 ["O", "O", "X", "O"]를 return합니다.

 */

 

import java.util.*;
public class programmer_0_12 {
    static String[] q1 = {"3 - 4 = -3", "5 + 6 = 11"};
    static String[] q2 = {"19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"};
 
    public String cal(String[] ar){
        String rst = "X";
        int cal_rst = 0;
        
        for(int i=0; i<ar.length; i++){
            switch(ar[1]){
                case "+" : 
                    cal_rst = Integer.parseInt(ar[0]) + Integer.parseInt(ar[2]);
                    break;
                case "-" :
                    cal_rst = Integer.parseInt(ar[0]) - Integer.parseInt(ar[2]);
                    break; 
            }
 
            if(Integer.parseInt(ar[4]) == cal_rst){
                rst = "O";
            }
        }
        return rst;
    }
    public String[] solution(String[] quiz) {
        String[] answer = new String[quiz.length];
 
        for(int i=0; i < quiz.length; i++){
            String[] ar = quiz[i].split(" ");
            answer[i] = cal(ar);
        }
        for(String i : answer){
            System.out.print(i + ",");
        }
        System.out.println();
 
        return answer;
    }
    public static void main(String args[]){
        programmer_0_12 t = new programmer_0_12();
        System.out.println("---------------------------------------");
        System.out.println("result = " + t.solution(q1));
        System.out.println("---------------------------------------");
        System.out.println("result2 = " + t.solution(q2));
        System.out.println("---------------------------------------");
    }
}