[백준/Baekjoon] 1935번: 후위 표기식2

1. 문제

후위 표기식과 각 피연산자에 대응하는 값들이 주어져 있을 때, 그 식을 계산하는 프로그램을 작성하시오.

2. 입력

첫째 줄에 피연산자의 개수(1 ≤ N ≤ 26) 가 주어진다. 그리고 둘째 줄에는 후위 표기식이 주어진다. (여기서 피연산자는 A~Z의 영대문자이며, A부터 순서대로 N개의 영대문자만이 사용되며, 길이는 100을 넘지 않는다) 그리고 셋째 줄부터 N+2번째 줄까지는 각 피연산자에 대응하는 값이 주어진다. (3번째 줄에는 A에 해당하는 값, 4번째 줄에는 B에 해당하는값 , 5번째 줄에는 C …이 주어진다, 그리고 피연산자에 대응 하는 값은 정수이다)

3. 출력

계산 결과를 소숫점 둘째 자리까지 출력한다.

4. 예제 입력 1

5
ABC*+DE/-
1
2
3
4
5

5. 예제 출력 1

6.20

6. 예제 입력 2

1
AA+A+
1

7. 예제 출력 2

3.00

8. 풀이

후위 표기식 문제와 이어지는 문제이다.

우선 알파벳(피연산자) 에 대응하는 값들을 리스트에 저장한다.

이후 주어진 후위 표기식을 변환하여 계산하게 된다. 이 과정에서 스택을 이용한다. 원리에 대한 설명은 다음과 같다.

if 'A' <= data[i] <= 'Z':
    result.append(alphabet[ord(data[i]) - ord('A')])

문자가 알파벳인 경우, 알파벳을 대응하는 값으로 변환하여 스택에 저장한다.

else:
    num2 = result.pop()
    num1 = result.pop()
    if data[i] == '+':
        result.append(num1 + num2)
    elif data[i] == '-':
        result.append(num1 - num2)
    elif data[i] == '*':
        result.append(num1 * num2)
    elif data[i] == '/':
        result.append(num1 / num2)

문자가 연산자일 경우, 스택의 맨 위의 값 2개를 꺼내 연산자에 따라 계산을 한 뒤, 다시 스택에 저장한다. 이 때 연산의 순서가 틀리지 않도록 주의해야 한다. 코드에서는 헷갈리지 않도록 꺼내는 순서와 반대로 변수의 이름을 정하였다.

모든 연산 과정이 끝나면 스택에 한 개의 값만 남게 된다. 이 값이 결과 값이다.

9. 코드

n = int(input())
data = input()

alphabet = [0] * n
for i in range(n):
    alphabet[i] = int(input())

result = []

for i in range(len(data)):
    if 'A' <= data[i] <= 'Z':
        result.append(alphabet[ord(data[i]) - ord('A')])
    else:
        num2 = result.pop()
        num1 = result.pop()
        if data[i] == '+':
            result.append(num1 + num2)
        elif data[i] == '-':
            result.append(num1 - num2)
        elif data[i] == '*':
            result.append(num1 * num2)
        elif data[i] == '/':
            result.append(num1 / num2)

print('%.2f' % result[0])