[프로그래머스] OX퀴즈(Java)

프로그래머스 - OX퀴즈

풀이

1. 들어온 문자열을 space 기준으로 배열을 만듦
2. 배열은 항상 [첫 번째 항, 연산자, 두 번째 항, =, 연산 결과값] 임
3. 첫 번째 항과 두 번째 항을 int로 바꾼 후, 연산자에 따라 연산 진행
4. 연산 결과와 실제 연산 결과가 같으면 O , 그렇지 않으면 X 반환

코드

import java.util.Arrays;

public class OXquiz {
	public String[] solution(String[] quiz) {
		String[] result = new String[quiz.length];
		for (int i = 0; i < quiz.length; i++) {
			String[] partition = quiz[i].split(" ");
			int sum = Integer.parseInt(partition[0]) + (partition[1].equals("+") ? 1 : -1) * Integer.parseInt(partition[2]);
			result[i] = sum == Integer.parseInt(partition[4]) ? "O" : "X";
		}
		return result;
	}

	public static void main(String[] args) {
		OXquiz oXquiz = new OXquiz();
		System.out.println(Arrays.toString(oXquiz.solution(new String[]{"3 - 4 = -3", "5 + 6 = 11"})));  // ["X", "O"]
		System.out.println(Arrays.toString(oXquiz.solution(new String[]{"19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"})));  // ["O", "O", "X", "O"]
	}
}