[ 2023-03-14 🌱 TIL ]

백준 1935번

입력으로 들어오는 후위 표기식을 주어진 입력값에 대입하여 답을 내는 문제입니다.

import sys

n = int(input()) # 입력으로 들어오는 수의 값
s = input().strip() # 후위 표기식을 입력으로 받는다
val = []

for _ in range(n):
    val.append(int(input()))

alphabet = {}
index1 = 0
for i in range(len(s)):
    ascii_s = ord(s[i]) # 아스키 코드로 변환하여 대문자에 그에 해당하는 값 딕셔너리에 저장
    if 65 <= ascii_s and ascii_s <= 90: 
        if s[i] not in alphabet:
            alphabet[s[i]] = val[index1] 
            index1 += 1

val_li = []
index2 = 0

for i in range(len(s)):
    if s[i] in alphabet: # 동일한 값이여도 배열에 넣는다.
        val_li.append(alphabet[s[i]]) 
    else: # 연산자가 등장했을 경우
    	# 연산자 이전의 두 개의 값을 추출하여 계산후 다시 배열에 넣는다 -> stack
        last = val_li.pop() 
        first = val_li.pop()
        # 순서 중요 나누기나 뺄셈같은 경우는 순서에 영향을 받기에
        value = 0
        if s[i] == '-':
            value = first - last
        elif s[i] == '*':
            value = first * last
        elif s[i] == '/':
            value = first / last
        elif s[i] == '+':
            value = first + last
        val_li.append(value)
        
print('{:.2f}'.format(val_li[0]))

입력으로 AA+A+ 같이 한가지의 값으로만 이루어진 후위 표기식이 들어올 때가 있어서

if s[i] in alphabet: # 동일한 값이여도 배열에 넣는다.
        val_li.append(alphabet[s[i]]) 

같은 부분으로 사용하여 해결하였습니다.

그러면 만약 A가 1인 입력값이라고 했을 때
배열에 처음 연산자가 나오기 전까지는 [ 1, 1 ]가 있습니다.
연산자 나오고 끝에 2개 값을 꺼내 계산 후 다시 배열에 넣어 [2] 가 됩니다.
그리고 다음 연사자가 나올 때까지 [2,1]이 들어가 있고 다시 위와 같은 일을 다시 진행하여 마지막에 배열에[3]이 남게 됩니다.