파이썬 - 전위 표현식

전위 표현식 확인용 사이트 : https://www.free-online-calculator-use.com/infix-to-prefix-converter.html

Infix to Prefix Converter | Interactive Step-By-Step Stack Tutorial

 

전위표현식은 후위 표현식과 유사하지만 입력받는 연산식을 뒤집어서 검사해주는 방식을 사용해준다.

처음에 입력받은 연산식을 뒤집고, 연산자들을 스택에 담고 피연산자는 바로 결과에 적어준다.

모든 동작이 끝나면 결과를 다시 뒤집어준다. 그 결과가 바로 전위표현식이 된다.

 

from collections import deque

def prefix_eval(inn):
  inn = inn[::-1]
  stack = deque()
  expression = ''
  math = ['+','-','*','/','(',')']
  for i in inn:
      
      if i not in math:
          expression+=i
      else:
          if i == math[2] or i== math[3]:
              stack.append(i)
          elif i == math[-1]:
              stack.append(i)
          elif i == math[4]:
              while stack:
                  out = stack.pop()
                  if out == ')':
                      break
                  expression += out
          elif i == math[0] or i == math[1]:
              if stack and stack[-1] not in ['+','-',')']:
                  while stack:
                      out = stack.pop()
                      if out in ['+','-',')']:
                          stack.append(out)
                          break
                      expression+=out
                  stack.append(i)
              else:
                  stack.append(i)
      # print('i :',i, '## st :',stack,'## ex :',expression)
  while stack:
      expression+=stack.pop()
  return expression[::-1]