[Python] 백준 1918 후위 표기식

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

1918번: 후위 표기식

 

문제 해결

- 하나씩 뺐을 때 문자/괄호/연산부호 분리해서 문자와 괄호,연산부호와 분리한다음 재배열을 해야한다. 어떻게 해야할까... => 문자는 그대로 출력하므로 뽑으면 바로 문자열에 이어 붙이고 괄호,연산부호는 stack 안에 넣은 후 배열 하는 것이 좋을 것이다.

- 이제 순서에 대해서 고민을 해야하는데 이는 그림으로 생각을 해야지 편하다.(그림은 직접 그려보길 추천한다. 여러가지 경우가 있고 해봐야지 감이온다.) 사실 몇 개 그렸는데 뭔가 아쉬워서 첨부하지 않는다..

 

 

CODE

import sys
input = sys.stdin.readline

cal = list(input().strip())
stack = []
r = ''
for spell in cal:
    if 'A' <= spell <='Z':
        r += spell   
    else:
        if spell == '(':
            stack.append(spell)
        elif spell == '*' or spell=='/':
            while stack and (stack[-1] == '*' or stack[-1]=='/'):
                r += stack.pop()
            stack.append(spell)
        elif spell == '+' or spell=='-':
            while stack and stack[-1] !='(':
                r += stack.pop()
            stack.append(spell)
        elif spell ==')':
            while stack and stack[-1] != '(':
                r += stack.pop()
            stack.pop()
            
while stack:
    r += stack.pop()
print(r)