[SWEA] 1233. [S/W 문제해결 기본] Tree - 사칙연산 유효성 검사

문제

https://www.swexpertacademy.com/main/learn/course/lectureProblemViewer.do

SW Expert Academy

 

 

 

풀이

이 문제는 두 자식 노드가 정수이고, 부모 노드가 연산자인지 확인한 후에 부모 노드에 연산 결과로 초기화해주는 과정을 반복하며 유효성을 검사하는 방식으로 구현했다.

 

 

import java.util.*;
 
public class Solution {
    static Scanner scan = new Scanner(System.in);
    static String[] nodes;
    static int num;
    static int chk = 1;
    static void solution() {
        for (int i = 1; i <= 10; i++) {
            num = scan.nextInt();
            nodes = new String[num+1];
            scan.nextLine();
            for(int j=1;j<=num;j++) {
                String tmp = scan.nextLine();
                nodes[j] = (tmp.split(" ")[1]);
            }
            test();
            System.out.println("#"+i+" "+chk);
            chk = 1;
        }
    }
    static void test() {
        int idx = num;
        while(idx!=1) {
            if(is_number(idx) && is_number(idx-1) && !is_number(idx/2)) {
                switch(nodes[idx/2]) {
                // 자식 노드가 모두 정수이고 부모가 연산자일 때, 부모의 연산자 요소에 따라 계산 후 부모 노드 초기화
                case "+":{
                    nodes[idx/2] = (Integer.parseInt(nodes[idx-1])+Integer.parseInt(nodes[idx])) +"";
                    break;
                }
                case "-":{
                    nodes[idx/2] = (Integer.parseInt(nodes[idx-1])-Integer.parseInt(nodes[idx])) +"";
                    break;
                }
                case "*":{
                    nodes[idx/2] = (Integer.parseInt(nodes[idx-1])*Integer.parseInt(nodes[idx])) +"";
                    break;
                }
                case "/":{
                    if(nodes[idx].equals("0"))//0으로 나누는 것은 고려하지 않고 유효성만 검사하므로 0 나누는 것이 나오면 1로 임시처리해줌
                        nodes[idx] = "1";
                    nodes[idx/2] = (Integer.parseInt(nodes[idx-1])/Integer.parseInt(nodes[idx])) +"";
                    break;
                }
                }
                idx-=2;
            }
            else {
                chk = 0;
                break;
            }
        }
    }
    static boolean is_valid(int idx) {
        if(idx>num)
            return false;
        return true;
    }
    static boolean is_number(int idx) {//숫자와 연산자를 구분하기 위한 함수
        if(!nodes[idx].equals("+") && !nodes[idx].equals("-") && !nodes[idx].equals("*") && !nodes[idx].equals("/"))
            return true;
        return false;
    }
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        solution();
    }
}