[Python] 백준 #1935- 후위 표기식2

문제

---

1935번: 후위 표기식2

코드

---

My answer

def calc(a,b,c):
    if(c=='+'):return a+b
    if(c=='-'):return a-b
    if(c=='*'):return a*b
    if(c=='/'):return a/b

import sys
input=sys.stdin.readline

n=int(input())
f=input().rstrip()
stack=[]
board=dict()

for i in f:
    if(i.isalpha() and i not in board):
        board[i]=int(input())
        
for i in f:
    if(i.isalpha()):
        stack.append(board[i])
    else:
        a=stack.pop()
        b=stack.pop()
        a,b=b,a
        stack.append(calc(a,b,i))
print("%.2f"%stack[-1])

Another answer

n = int(input())
expression = input()
numbers = [int(input()) for i in range(n)]

stk = []
ans = 0
for x in expression:
    if x.isalpha():
        stk.append(numbers[ord(x) - ord('A')])
    else:
        b = stk.pop()
        a = stk.pop()
        stk.append(eval(f'{a} {x} {b}'))

print(f'{stk.pop():.2f}')

풀이

---

후위 표기식을 몰라서 인터넷에 찾아보고 공부했다. 혹시라도 모를 분들을 위해서 알려주면 사칙연산이랑 똑같은 순서로 하게끔 식을 써놓는 것이다. 예를 들어 ABC*+DE/-는 우선 *가 젤 앞에있으므로 B*C 그리고+가 나오므로 B*C+A가 되고 DE다음 /가 있으므로 D/E 그리고 맨마지막 -까지하면 (B*C)+A-D/E가 된다. 

식을 이해하고 보니 그냥 바로 스택이 떠올라서 스택으로 알파벳이 나오면 스택에 저장하고, 연산기호가 나오면 스택맨위에서 2개를 꺼내 계산한 뒤 결과값을 다시 스택에 넣어줬다. 다만 여기서 주의할 점은 스택은 거꾸로 넣어지기 때문에 꺼낼때 +는 앞뒤순서가 상관없지만 (-.*./)는 순서가 중요하므로 스택에서 꺼낸뒤 그 수들의 순서를 바꿔줘야한다. 무슨말이냐면 우슨 스택에 [A,B,C]가 저장되고 '-'기호가 들어왔다고 할 때 스택에서 꺼내면 C,B,순서가 되는데 계산은 B-C를 해야하므로 순서를 바꿔줘야 한다는 것이다. 그리고 나는 연산자가 문자형이므로 계산을 하려고 따로 함수를 만들어줬는데 이미 파이썬에 eval()이라는 함수가 있다고 한다... 아래 코드는 그 함수를 쓴 코드이다. 다음에 알고리즘 카테고리에 eval()에 대해서 작성하도록 하겠당