후위 표기식2

문제

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

입력

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

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

출력

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

예제 입력 1 

5
ABC*+DE/-
1
2
3
4
5

예제 출력 1 

6.20

예제 입력 2 

1
AA+A+
1

예제 출력 2

3.00

 

import java.util.*;
import java.io.*;
 
public class Main {
    public static void main(String[] args) throws IOException {
        //TIP Press <shortcut actionId="ShowIntentionActions"/> with your caret at the highlighted text
        // to see how IntelliJ IDEA suggests fixing it.
       BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
       int n = Integer.parseInt(br.readLine());
 
       String expression = br.readLine();
 
        String[] value = new String[n];
 
       for(int i = 0; i < n; i++) {
            value[i] = br.readLine();
       }
 
       double result = solution(expression, value);
 
       System.out.printf("%.2f", result);
    }
 
    public static double solution(String expression, String[] input) {
        Stack<Double> stack = new Stack<>();
        for(int i = 0; i < expression.length(); i++) {
            char c = expression.charAt(i);
            if(c >= 'A' && c <= 'Z') {
                stack.push(Double.parseDouble(input[c - 'A']));
            } else {
                double b = stack.pop();
                double a = stack.pop();
                if(c == '+') stack.push(a + b);
                if(c == '-') stack.push(a - b);
                if(c == '*') stack.push(a * b);
                if(c == '/') stack.push(a / b);
            }
        }
        return stack.pop();
    }
}
 
 

 

후위연산자에 대해 설명하면 연산자가 피연산자 뒤에 가는 방식을 말한다.

문제를 어떻게 풀어야할지 고민하는데 stack이 나왔다.피연산자를 담아뒀다가 연산자가 나오면 담아둔 연산자를 연산한다. 

이 문제는 숫자가 아니라 문자라 오히려 쉬웠다. 십의자리수가 나올때는 어떻게 해야할지 잘 몰랐는데 문자라 그런 걸 걱정할 이유가 없었다. 연산자가 나오면 스택에서 피연산자를 가지고와 계산한 다음 push 해주어 계속 계산해나가면 되는 문제였다.