후위 표기식2 풀이

문제 링크: https://www.acmicpc.net/problem/1935

백준 알고리즘 기초 1/2 203에서 1번째 - 1935번 후위 표기식2를 풀어보았다.

 

풀이: 스택에 수를 넣고 계산하는 문제이다.

 

C++

 

Python

n=int(input())
inp=input()
nums=[int(input()) for i in range(n)]
stack=[]
for i in range(len(inp)):
	if inp[i].isalpha(): stack.append(nums[ord(inp[i])-65]) # 수 집어넣기
	else:
		a=stack.pop() # 수 계산
		calc=stack.pop()
		if inp[i]=='+': calc+=a
		elif inp[i]=='-': calc-=a
		elif inp[i]=='*': calc*=a
		elif inp[i]=='/': calc/=a
		stack.append(calc)
print("%.2f"%stack[0]) # 소수점 출력

Java