[백준] 15815 - 천재 수학자 성필

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

15815번: 천재 수학자 성필

동아리 알고리즘 스터디의 멘토로서, 스택 연습문제 추천을 위해 스택 문제를 찾던 중

후위연산식 문제와 유사한 문제를 발견하여 풀어보았다.

 

후위연산식 문제가 중위연산식 문제를 후위연산식으로 고치는 문제라면

이 문제는 후위연산식을 보고 그 값을 계산한 결과를 출력하는 문제이다.

 

처음에는 소수점 부분에 이상하게 얽메여서 삽질을 좀 했지만,

중간 계산 결과가 항상 정수라는 점을 유의하여 나눗셈계산을 하면 된다.

(c언어 계열의 경우 int형을 쓰면 사실 따로 생각할 것이 없긴 하지만,

파이썬의 경우 자동으로 소수점연산을 하므로 주의해야한다.)

 

from collections import deque

s = input()
d = deque()
for i in range(len(s)):
    try:
        d.append(int(s[i]))
    except:
        if s[i] == '+':
            answer = (d.pop() + d.pop())

        elif s[i] == '-':
            op1 = d.pop()
            op2 = d.pop()
            answer = (op2-op1)

        elif s[i] == '*':
            answer = (d.pop() * d.pop())

        elif s[i] == '/':
            op1 = d.pop()
            op2 = d.pop()
            answer = op2 // op1

        d.append(answer)
print(d.pop())