[알고리즘][백준] 1918번 후위 표기식 - 파이썬(python) 풀이

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

1918번: 후위 표기식

---

1935번 후위 표기식2와 비슷하지만 더 어려웠다...

 

참고해서 만든 나의 코드

 

이번에는 stack에 연산자만 넣고 진행하게 되는데 그 생각이 쉽게 나지 않았다...  

import sys

input = sys.stdin.readline().rstrip()

input = "A*(B+C)"

answer = ""

stack = []  # 연산자만 넣음

for i in input:

    # (,) -> *,/ -> +-

    if not i.isalpha():

        # 괄호 시작

        if i == "(":

            stack.append(i)

        # *,/

        elif (i == "*") or (i == "/"):

            # stack 존재 & A*B*C -> AB*C*

            while stack and (stack[-1] == "*" or stack[-1] == "/"):

                answer += stack.pop()

            stack.append(i)

        # +-

        elif (i == "+") or (i == "-"):

            # stack 존재 & *,/,+,- 경우 pop

            while stack and stack[-1] != "(":

                answer += stack.pop()

            stack.append(i)

        # 괄호 끝

        elif i == ")":

            # +-*/ pop & (도 pop

            while stack and stack[-1] != "(":

                answer += stack.pop()

            stack.pop()

    # 알파벳

    else:

        answer += i

while stack:

    answer += stack.pop()

print(answer)