[JAVA] ★ OX퀴즈 split()

덧셈, 뺄셈 수식들이 '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) indexOf() 와 substring()을 이용한 풀이

class Solution {
    public String[] solution(String[] quiz) {
        String[] answer = new String[quiz.length];
        
        for(int i =0;i<quiz.length;i++){
           int a = quiz[i].indexOf("=");
           // 결과값
           int sum = Integer.parseInt(quiz[i].substring(a+2));
           // " = 결과값" 제거
           String qui = quiz[i].replace(" = "+sum,"");
           
           if(qui.indexOf("+")!=-1){
               int b = qui.indexOf("+");
               // 2번째 숫자
               int num2 = Integer.parseInt(qui.substring(b+2));
               // 1번째 숫자 (수식과 2번숫자 제거)
               int num1 = Integer.parseInt(qui.replace(" + "+num2,""));
               int sum1 = num1 + num2;
                  if(sum1 == sum){
                      answer[i] ="O";
                  } else {
                      answer[i] ="X";
                  }
           } else {
               int b = qui.indexOf("-");
               int num2 = Integer.parseInt(qui.substring(b+2));
               int num1 = Integer.parseInt(qui.replace(" - "+num2,""));
               int sum1 = num1 - num2;
                  if(sum1 == sum){
                      answer[i] ="O";
                  } else {
                      answer[i] ="X";
                  }
           }    
       }
       return answer;
    }
}

몇가지의 테스트의 런타임 오류로 인해 재풀이

풀이2) split()를 활용한 풀이

class Solution {
    public String[] solution(String[] quiz) {
        String[] answer = new String[quiz.length];
        for(int i =0; i<quiz.length;i++){
        	//공백을 구분으로 나눔
            String[] arrays = quiz[i].split(" ");
            
            if(arrays[1].equals("+")){
                int num1 = Integer.parseInt(arrays[0]);
                int num2 = Integer.parseInt(arrays[2]);
                int sum = num1 + num2; 
                int result = Integer.parseInt(arrays[4]);
                if(result==sum){
                    answer[i]="O";                    
                } 
                else {
                    answer[i]="X";
                }
            } else{
                int num1 = Integer.parseInt(arrays[0]);
                int num2 = Integer.parseInt(arrays[2]);
                int sum = num1 - num2; 
                int result = Integer.parseInt(arrays[4]);
                if(result==sum){
                    answer[i]="O";                    
                } 
                else {
                    answer[i]="X";
                }
            }  
        }
        return answer;
    }
}

split("구분자");

결과는 정답으로 나왔지만, 코드가 너무 길어지는 문제점이나 몇 가지 문제로 인해

간결하게 표현하는 다른 풀이 처럼 연습이 필요할 것 같다.

풀이 3)  split 와 삼항연산자 등을 활용한 풀이 

class Solution {
  public String[] solution(String[] quiz) {
    for(int i=0; i<quiz.length; i++){
     String[] text = quiz[i].split(" ");
     int result = Integer.parseInt(text[0]) + ( Integer.parseInt(text[2]) * ( text[1].equals("+") ? 1:-1) );
     //삼항연산자를 활용해 +, -를 풀어내어 간결하게 표시
     quiz[i] = result == Integer.parseInt(text[4])? "O": "X";
     //answer의 변수를 활용하지 않고 quiz를 재활용하여 return
     }
      return quiz;
  }
}