[백준 1935번] 후위표기식2 :: 늦깎이 IT

[백준 1935번] 후위표기식2 URL

문제풀이는 후위표기식과 후위표기식 계산에 대한 아래의 포스팅을 참고하시기 바랍니다.

2019/03/21 - [알고리즘 이론] - [자료구조] 스택으로 후위표기식으로 변환하기 :: 늦깎이 IT

2019/03/21 - [알고리즘 이론] - [자료구조] 스택으로 후위표기식 계산하기 :: 늦깎이 IT

[소스 코드]

 
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Scanner;
import java.util.Stack;
 
public class Main {
 
    static int N;
    static int[] arr;
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
    public static void main(String[] args) throws NumberFormatException, IOException {
 
        N = Integer.parseInt(br.readLine());
        String str = br.readLine();
        arr = new int[N];
 
        for (int i = 0; i < N; i++)
            arr[i] = Integer.parseInt(br.readLine());
 
        System.out.printf("%.2f", calPostfix(str));
 
    }
 
    public static double calPostfix(String input) {
 
        Stack<Double> stack = new Stack<>();
        int len = input.length();
 
        for (int i = 0; i < len; i++) {
 
            char op = input.charAt(i);
 
            if (op >= 'A' && op <= 'Z') {
                stack.push((double) arr[op - 'A']);
            } else {
 
                double op2 = stack.pop();
                double op1 = stack.pop();
                switch (op) {
 
                case '+':
                    stack.push(op1 + op2);
                    break;
                case '-':
                    stack.push(op1 - op2);
                    break;
                case '*':
                    stack.push(op1 * op2);
                    break;
                case '/':
                    stack.push(op1 / op2);
                    break;
                }
            }
        }
        return stack.pop();
    }
}