백준: 1935 후위 표기식2

문제

1935번: 후위 표기식2

 

문제 풀이

 후위 표기식을 계산하는 것은 후위 표기식을 만드는 것보다는 간단하다. 피연산자를 stack에 push 하고, 연산자가 입력되면 stack의 값 중 2개를 출력하여 연산을 하면 된다. 해당 문제에서는 A - Z와 같이 영어 대문자가 주어지고, 이에 매칭 되는 숫자들이 입력될 때 후위 표기식을 계산하는 문제이다.

 

• `string.ascii_uppercase`를 활용하여 A - Z에 대응되는 딕셔너리를 초기화 한다.
  • `stirng.ascii_uppercase`는 사전에 정의된 string 상수로 ABCDEFGHIJKLMNOPQRSTUVWXYZ를 의미한다.
• 입력된 값이 알파벳인 경우 stack에 push하고 연산자가 입력된 경우 계산을 수행하여 다시 stack에 추가한다.

 

코드

from sys import stdin
import string

if __name__ == "__main__":
    change = dict()
    stack = []
    num_operand = int(stdin.readline())
    math_expr = stdin.readline()[:-1]

    # 알파벳과 매칭되는 숫자를 찾기 위해 딕셔너리 생성
    for alpha in string.ascii_uppercase[:num_operand]:
        change[alpha] = int(stdin.readline())

    for c in math_expr:
        if c.isalpha():
            stack.append(change[c])
        else:
            n2, n1 = stack.pop(), stack.pop()
            if c == '*':
                stack.append(n1 * n2)
            elif c == '/':
                stack.append(n1 / n2)
            elif c == '+':
                stack.append(n1 + n2)
            else:
                stack.append(n1 - n2)

    print("%.2f" % stack[-1])