프로그래머스 120907. OX퀴즈 (JAVA)

1. 문제

https://school.programmers.co.kr/learn/courses/30/lessons/120907

2. 풀이

직전에 풀었던 풀이와 동일한 스타일로 풀었다. 

 

가장 일반적인 접근은 공백(" ")으로 split을 한후 고정된 위치의 값을 얻고, 비교하면 것이다.

 

내 답안의 코드를 보면 알 수 있지만, 런타임 에러가 나는 경우는

15 - -5 = 10

위와 같은 테스트케이스가 들어오면 런타임 에러가 나니 주의가 필요하다.

 

개인적으로 스트림을 좋아하는데 인상깊었던 답안은 진짜 일반적인 자바코드에서 보기 어려운 길고긴 스트림을 썼다. 나도 저렇게 쓸 수 있는 날이 올까?

3. 코드

내 답안

import java.util.*;

class Solution {
    public String[] solution(String[] quiz) {

        ArrayList<String> answerList = new ArrayList<>();
        for (String q : quiz) {
            String[] quizSplit = q.replaceAll("- -", "").replaceAll("- ", "-").replaceAll("[+] ", "").split(" ");
            if (Integer.parseInt(quizSplit[0]) + Integer.parseInt(quizSplit[1]) == Integer.parseInt(quizSplit[3])) {
                answerList.add("O");
            } else {
                answerList.add("X");
            }
        }
        String[] answer = answerList.toArray(new String[0]);
        return answer;
    }
}

 

인상깊었던 답안(홍희표)

import java.util.Arrays;

class Solution {
    public String[] solution(String[] quiz) {
        return Arrays.stream(quiz).map(s -> {
            String[] arr = s.trim().split(" ");
            return arr[1].equals("+") && Integer.parseInt(arr[0]) + Integer.parseInt(arr[2]) == Integer.parseInt(arr[4]) || Integer.parseInt(arr[0]) - Integer.parseInt(arr[2]) == Integer.parseInt(arr[4]) ? "O" : "X";
        }).toArray(String[]::new);
    }
}