1935 후위 표기식2

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

1935번: 후위 표기식2

 

내 풀이(맞음)

n = int(input())
x = input()
rst = []
for i in range(n):
    a = int(input())
    rst.append(a)
index = []
for i in range(len(x)):
    if x[i] == '+' or x[i] == '-' or x[i] == '*' or x[i] == '/':
        index.append(x[i])
    else:
        index.append(rst[ord(x[i])-ord('A')])
ary = []
for i in index:
    if i == '+' or i == '-' or i == '*' or i == '/':
        t1 = ary.pop()
        t2 = ary.pop()
        t3 = 0
        if i == '+':
            t3 = t2 + t1
        elif i == '-':
            t3 = t2 - t1
        elif i == '*':
            t3 = t2 * t1
        else:
            t3 = t2 / t1
        ary.append(t3)
    else:
        ary.append(i)
print('%.2lf'%ary[0])

rst에 숫자를 입력받는다. 그리고 for문에서 사칙연산이 아니면 index.append(rst[ord(x[i])-ord('A')]) 이렇게 한다. 왜냐하면 저래야 A, B, C 순서대로 rst의 저장된 값으로 대체되기 때문이다. 그리고 스택을 이용해 계속 넣다가 사칙연산을 만나면 위의 두 숫자를 연산해주고 다시 ary에 넣어준다.