[Java Algorithm] 사칙연산 유효성 검사 SWEA #1233

문제

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV141176AIwCFAYD 

SW Expert Academy

 

내 문제 풀이

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Solution {
 
    public static void main(String[] args) throws Exception {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuffer out = new StringBuffer();
         
        for(int t = 1; t <= 10; t++) {
            int n = Integer.parseInt(in.readLine());
            int answer = 1;         
             
            for(int i = 0; i < n; i++) {
                st = new StringTokenizer(in.readLine());
                if(answer == 0) continue; // answer가 0이면 다음 입력부터 검사할 필요x
                st.nextToken(); // 노드 번호 알 필요x => 패스
                String cal = st.nextToken();
                 
                if(st.hasMoreTokens()) { // 입력 남았을 떼 == 단말 노드가 아닐 때 == 숫자면 안돼 !
                    if(!(cal.equals("+") || cal.equals("-") || cal.equals("*") || cal.equals("/"))) answer = 0;
                } else { // 더 이상 입력 없을 때 == 단말 노드일 때 == 연산자면 안돼 !
                    if(cal.equals("+") || cal.equals("-") || cal.equals("*") || cal.equals("/")) answer = 0;
                }
            }
            out.append("#" + t + " " + answer + "\n");
        }
        System.out.println(out);
    }
}

👉 구현

연산이 성립되고 안되고의 여부를 판단하는 기준이 굉장히 간단하기 때문에

굳이 그래프를 이용하여 노드를 따라가는 등의 동작을 할 필요 없이, 입력받으면서 바로 확인을 했다.

 

✔️ 연산이 성립되지 않는 기준

1) 단말 노드는 연산자면 안된다.

2) 단말 노드가 아닐 경우 숫자면 안된다.

 

[ 풀이 설명 ]

• n번 반복하면서 일단 해당 입력 라인을 받아온 후, answer을 확인했다.
  입력을 받으면서 연산이 성립되지 않음을 확인하면 바로 answer를 0으로 바꿔줄 것이므로
  answer가 0이면 더이상 연산 성립 여부를 확인할 필요가 없다.
• answer가 0이 아니라면 일단 입력된 값들을 확인해야 하는데,
  입력된 값들 중 첫 번째 값인 노드 번호는 필요가 없으므로 받아서 날려버렸다.
• 두 번째 값은 해당 노드에 들어가는 값이므로, 연산자인지 숫자인지 판단해주어야 한다.
  따라서 cal변수로 입력받고, 조건문으로 위에서 말한 연산이 성립되지 않는 기준을 확인했다.
• hasMoreTokens() 메서드를 이용하여 입력이 남은지 확인, 단말노드인지 아닌지를 판단했고,
  (1)단말노드가 아닌데 연산자가 아닐 경우, (2)단말노드인데 연산자인경우 두 경우에
  더이상 이 트리는 연산이 성립되지 않는다고 보아 answer을 0으로 만들어주었다.

 

💡 피드백

• 입력값이 넘나 복잡해보여서 문제 이해에 시간을 많이 썼지만,,,! 쉬운 문제였다.