[BAEKJOON] 1935 | 후위 표기식 2

🔎 1935 | 후위 표기식 2

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

---

💡 Solution

✍🏻

Logic

1. 후위표기식을 토대로 연산을 수행하는 문제이다.

2. 먼저 알파벳과 들어오는 숫자를 매칭 시킨다.

3. 이후, 후위 표기식대로 연산을 수행한다.

4. 피연산자가 들어올 경우, 알파벳에 해당하는 숫자를 스택에 push 한다.

5. 연산자가 들어올 경우, 스택에 있는 두 연산자를 가지고 와서 연산을 수행한 후, 결과를 다시 스택에 push한다.

n = int(input().rstrip())
s = input().rstrip()
num_list = {}

for i in range(1, n+1) :
    num_list.update({chr(64 + i) : int(input().rstrip())})
    
def calculate(s) :
    stack = []
    for i in range(len(s)) :
        if ord('A') <= ord(s[i]) and ord(s[i]) <= ord('Z') :
            stack.append(num_list[s[i]])
        else :
            b = stack.pop()
            a = stack.pop()
            
            if s[i] == '+' :
                stack.append(a+b)
            elif s[i] == '-' :
                stack.append(a-b)
            elif s[i] == '*' :
                stack.append(a*b)
            elif s[i] == '/' :
                stack.append(a/b)
                
    return float(stack.pop())

result = "{:.2f}".format(calculate(s))
print(result)

---

▶ 📌 Note

∙ 설명
소수점 n자리 형식 맞추기에는 f-string 방식과 format 방식이 있다.

dictionary를 사용했기 때문에 굳이 아스키코드로 비교하지 않고 'in'을 사용하면 더 효율적인 코드가 될 수 있다.

○ f-string
- f"{number:.2f}"

○ format
- "{:.2f}".format(number)