[백준] 1918번 파이썬 - 스택

문제요약 (1918. 후위표기식)

 

1918번: 후위 표기식

 

 

풀이를 위한 부연 설명

• 연산자의 우선순위를 기준으로 스택에 쌓아야한다.
• 괄호 처리에 유의해야한다.

 

• 스택에 대한 자세한 개념 설명과 연관 문제는 다음 링크를 참고하면 된다.

[구버전] [Python] 03. 스택과 큐

 

 

코드 

import sys
input = sys.stdin.readline

exp = input().rstrip()

stack = []
ans = ""
idx = 0

while idx < len(exp):
    e = exp[idx]
    if e == "(":
        stack.append(e)
    elif e == ")":
        while stack and stack[-1] != "(":
            ans += stack.pop()
        stack.pop()
    elif e == "*" or e == "/":
        while stack:
            if stack[-1] == "*" or stack[-1] == "/":
                ans += stack.pop()
            else:
                break
        stack.append(e)
    elif e == "+" or e == "-":
        while stack and stack[-1] != "(":
            ans += stack.pop()
        stack.append(e)
    else:
        ans += e
    idx += 1

while stack:
    ans += stack.pop()

print(ans)

 

 

 

기억할 점

• 괄호 없는 후위 표기식은 만들기 쉬웠는데 이후가 어려웠다