BOJ01935 후위 표기식2

https://www.acmicpc.net/problem/1935 (opens new window)

후위 표기식과 각 피연산자에 대응하는 값들이 주어져 있을 때, 그 식을 계산한다.

WARNING

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

후위 표기식

ABC*+DE/-을 후위 표기식으로 표현하면 아래와 같이 표현할 수 있다.

(A + (B * C)) - (D / E) 

스택

// TODO

Stack<Double> stack = new Stack<>();
for (char c : postfix.toCharArray()) { // (1)
    if (Character.isAlphabetic(c)) { // (2)
        stack.push(Double.valueOf(map.get(c)));
    } else { // (3)
        double a = stack.pop();
        double b = stack.pop();
        if (c == '+') {
            stack.push(b + a);
        } else if (c == '-') {
            stack.push(b - a);
        } else if (c == '*') {
            stack.push(b * a);
        } else {
            stack.push(b / a);
        }
    }
}

(1) 입력 받은 후위연산자를 모두 순회하기 위한 반복문이다.
(2) 만약 c가 알파벳인 경우 stack에 push를 진행한다.
(3) 연산자인 경우 연산 진행을 위해 피연산자를 꺼내 연산 후 다시 push 한다.

제출 코드

아래는 최종 제출 코드이다.

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.Stack;

public class BOJ1935 {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        int n = scanner.nextInt(); // 피연산자의 개수 1 <= n <= 26
        String postfix = scanner.next();

        Map<Character, Integer> map = new HashMap<>();

        for (int i = 0; i < n; i++) {
            map.put((char) ('A' + i), scanner.nextInt());
        }

        Stack<Double> stack = new Stack<>();
        for (char c : postfix.toCharArray()) {
            if (Character.isAlphabetic(c)) {
                stack.push(Double.valueOf(map.get(c)));
            } else {
                double a = stack.pop();
                double b = stack.pop();
                if (c == '+') {
                    stack.push(b + a);
                } else if (c == '-') {
                    stack.push(b - a);
                } else if (c == '*') {
                    stack.push(b * a);
                } else {
                    stack.push(b / a);
                }
            }
        }

        System.out.printf("%.2f", stack.pop());
    }
}

#BOJ #자료 구조 #스택