[프로그래머스]OX퀴즈

OX퀴즈

문제 설명

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

---

입출력 예 설명

입출력 예 #1

• 3 - 4 = -3 은 틀린 수식이므로 “X”, 5 + 6 = 11 은 옳은 수식이므로 “O” 입니다. 따라서 [“X”, “O”]를 return합니다.

입출력 예 #2

• 19 - 6 = 13 은 옳은 수식이므로 “O”, 5 + 66 = 71 은 옳은 수식이므로 “O”, 5 - 15 = 63 은 틀린 수식이므로 “X”, 3 - 1 = 2는 옳은 수식이므로 “O” 따라서 [“O”, “O”, “X”, “O”]를 return합니다.

---

나의 풀이 코드

import java.util.*;
class Solution {
    public String[] solution(String[] quiz) {
        String[] answer = {};
        
        List<String> result_list = new ArrayList<>(); 
        
        for(int i=0; i<quiz.length; i++){
            
            List<String> list = Arrays.asList((quiz[i]).split(" "));
            int cal=Integer.parseInt(list.get(0));
            for(int j=0; j<list.size(); j++){
                
                if (list.get(j).equals("=")){
                    
                    if (list.get(j+1).equals(""+cal)){
                        result_list.add("O");
                    }
                    else if (!list.get(j+1).equals(""+cal)){
                        result_list.add("X");
                    }
                    break;
                }
                else if (list.get(j).equals("+")){
                    
                    cal += Integer.parseInt(list.get(j+1));
                    
                }
                else if (list.get(j).equals("-")){
                    
                    cal -= Integer.parseInt(list.get(j+1));
                }
                
                
                
            }
            
        }
        
        answer = result_list.toArray(new String[result_list.size()]);
        return answer;
    }
}

반복문안에서 =,+,-를 판별하여 계산후 결과값에 맞는지 판별했다. 또한 어제 배운 정수를 스트링으로 쉽게 바꾸는법을 사용했다.

배운점

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";
        }
        return quiz;
    }
}

다른사람이 풀은 코드이다.
삼항 연산자를 통해서 코드를 20줄을 2줄로 줄였다.
삼항 연산자에 익숙해지도록 연습하자.