(24.01.01) (Baekjoon) 1935. 후위 표기식2

 

1935번: 후위 표기식2

 

시간 제한 : 2초

메모리 제한 : 128mb

 

문제

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

입력

첫째 줄에 피연산자의 개수(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.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.ArrayList;

public class Main {
	
	static Stack<Double> stack_num = new Stack<Double>();
	public static void main(String[] args) throws Exception {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int n = Integer.parseInt(br.readLine());
		char[] ar = br.readLine().toCharArray();
		double[] nums = new double[n];
		for (int i = 0; i < n; i++) {
			nums[i] = Double.parseDouble(br.readLine());
		}
		for (int i = 0; i < ar.length; i++) {
//			System.out.println(ar[i]);
			if((ar[i] >= 'A' && ar[i] <= 'Z')) {
				stack_num.push(nums[ar[i]-'A']);
			}
			else {
				double a = stack_num.pop();
				double b = stack_num.pop();
				double res = 0.00;
				switch(ar[i]) {
					case '+':
						res = b+a;
						break;
					case '-':
						res = b-a;
						break;
					case '*':
						res = b*a;
						break;
					case '/':
						res = b/a;
						break;
				}
				stack_num.push(res);
			}
		}
		
		System.out.printf("%.2f", stack_num.pop());
	}

}