백준 알고리즘 1935번 후위 표기식2(python)

스택 문제이다.

N = int(input())
li = list(input())
nums = [int(input()) for _ in range(N)]
output = []
for t in li:
    if t in "+-*/":
        a = output.pop()
        b = output.pop()
        if t == '+':
            output.append(b+a)
        elif t == '-':
            output.append(b-a)
        elif t == '*':
            output.append(b*a)
        elif t == '/':
            output.append(b/a)
    else:
        output.append(nums[ord(t)-ord('A')])
print("%.2f" % (output[0]))