[백준] 1935: 후위 표기식 2 (JAVA)

 

문제

후위 표기식과 각 피연산자에 대응하는 값들이 주어져 있을 때, 그 식을 계산하는 프로그램을 작성하시오.

입력

첫째 줄에 피연산자의 개수(1 ≤ N ≤ 26) 가 주어진다. 그리고 둘째 줄에는 후위 표기식이 주어진다. (여기서 피연산자는 A~Z의 영대문자이며, A부터 순서대로 N개의 영대문자만이 사용되며, 길이는 100을 넘지 않는다) 그리고 셋째 줄부터 N+2번째 줄까지는 각 피연산자에 대응하는 값이 주어진다. 3번째 줄에는 A에 해당하는 값, 4번째 줄에는 B에 해당하는값 , 5번째 줄에는 C ...이 주어진다, 그리고 피연산자에 대응 하는 값은 100보다 작거나 같은 자연수이다.

후위 표기식을 앞에서부터 계산했을 때, 식의 결과와 중간 결과가 -20억보다 크거나 같고, 20억보다 작거나 같은 입력만 주어진다.

출력

계산 결과를 소숫점 둘째 자리까지 출력한다.

 

풀이

피연산자를 어떻게 처리해야 할까를 가장 오래 고민했다. 피연산자 배열을 따로 만들고, 아스키 코드를 사용한 수식으로 피연산자 배열에서 꺼내서 사용했다.

 

정답 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int N = Integer.parseInt(br.readLine());

        double[] arr = new double[N]; // 피연산자 담을 배열
        Stack<Double> stack = new Stack<>(); // 연산값 담을 스택

        String s = br.readLine();

        for (int i = 0; i < N; i++) {
            arr[i] = Double.parseDouble(br.readLine()); // 피연산자 담기
        }

        for (int j = 0; j < s.length(); j++) {
            if ('A' <= s.charAt(j) && s.charAt(j) <= 'Z') {
                stack.push(arr[s.charAt(j) - 'A']); // ex) arr[0]에 A가 들어있으므로 A-A를 하면 0, 인덱스처럼 사용 가능
            } else if (s.charAt(j) == '+') {
                stack.push(stack.pop() + stack.pop());
            } else if (s.charAt(j) == '-') {
                double tmp1 = stack.pop();
                double tmp2 = stack.pop();
                stack.push(tmp2 - tmp1);
            } else if (s.charAt(j) == '*') {
                stack.push(stack.pop() * stack.pop());
            } else if (s.charAt(j) == '/') {
                double tmp1 = stack.pop();
                double tmp2 = stack.pop();
                stack.push(tmp2 / tmp1);
            }
        }
        double res = stack.pop();
        System.out.println(String.format("%.2f", res));
    }
}

 

알게된 점

• 아스키코드 적용법
• String.format 사용법