[백준/파이썬]1935 후위 표기식2

https://www.acmicpc.net/problem/1935

1935번: 후위 표기식2

 

문제

 

후위표기식을 계산하면 되는 문제다.

 

후위표기식 계산방법은 연산자가 나왔을 때 연산자 앞에 있는 숫자 두 개를 계산하면 된다.

 

이 문제에서는 알파벳과 숫자가 따로 주어지는데, 알파벳에 숫자를 대입해주는 것만 신경 쓰면 쉽게 풀 수 있는 문제였던 것 같다.

 

정답

n = int(input())

s = input()
arr_dict = {}
for i in range(n) :
    arr_dict[chr(65+i)] = int(input())

stack = []

for c in s :
    if c == '+' or c == '-' or c == '*' or c == '/' :
        n1 = stack.pop()
        n2 = stack.pop()
        if c == '+' :
            stack.append(n2 + n1)
        elif c == '-' :
            stack.append(n2 - n1)
        elif c == '*' :
            stack.append(n2 * n1)
        elif c == '/' :
            stack.append(n2 / n1)
    else:
        stack.append(arr_dict[c])
        
print('{:.2f}'.format(stack[0]))