[Baekjoon-Python] 1918 : 후위 표기식

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

접근

여는 괄호와 연산자는 스택에 삽입한다.
여는 괄호는 닫는 괄호를 만났을 때 스택에서 삭제한다.
연산자는 다음 연산자보다 우선순위가 높을 때 스택에서 삭제한다.

풀이

from sys import stdin

infix = stdin.readline().rstrip()
priority = {"*": 2, "/": 2, "+": 1, "-": 1}
stack = []
res = []
for token in infix:
    if 65 <= ord(token) <= 90: # (1)
        res.append(token)
    else:
        if token == ")":
            while stack[-1] != "(": # (2)
                res.append(stack.pop())
            stack.pop() # (3)
        elif token == "(":
            stack.append(token) # (4)
        else:
            if stack and stack[-1] != "(" and priority[stack[-1]] >= priority[token]: # (5)
                while stack and stack[-1] != "(" and priority[stack[-1]] >= priority[token]:
                    res.append(stack.pop())
            stack.append(token)

while stack:
    res.append(stack.pop()) # (6)

print("".join(res))