[프로그래머스 코딩테스트] OX 퀴즈(Java)

  문제 설명

  덧셈, 뺄셈 수식들이 '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"]

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

  answer 배열의 크기는 quiz의 길이로 지정합니다.

  for문이 quiz의 길이만큼 순회합니다.

  배열 s는 split 메소드를 이용하여 quiz의 i 번째 원소를 공백으로 구분한 값을 원소로 갖습니다.

  계산한 값을 담을 변수 cal을 생성합니다.

  바깥 if문에서 s의 1번 인덱스, 즉 연산자가 +인지 확인합니다.

  cal에 s의 0번 인덱스와 2번 인덱스의 값끼리 더한 값을 할당합니다. 더하기 전 Integer 클래스의 parseInt 메소드를 이용하여 문자열이던 수를 정수형으로 형변환합니다.

  안쪽 if문에서 cal이 s의 4번 인덱스의 값과 동일하다면 answer의 i 번째 원소에 O를, 아니라면 X를 할당합니다.

  마찬가지로 s의 1번 인덱스인 연산자가 -인지 확인합니다.

  cal에 s의 0번 인덱스와 2번 인덱스의 값끼리 뺀 값을 할당합니다.

  두 번째 안쪽 if문에서 cal이 s의 4번 인덱스의 값과 동일하다면 answer의 i 번째 원소에 O를, 아니라면 X를 할당합니다.

 

  문제에서 요구한 사항을 그대로 담고 있는 풀이입니다. 가능하다면 삼항 연산자 등을 이용하여 좀 더 간결하게 풀이해나갈 수 있다면 좋을 테지만, 아직 코드를 깔끔하게 작성하는 것은 어렵네요.