Algorithm (PS)

[백준] 1935 후위표기식 2 in Python

minjiwoo 2022. 9. 19. 13:56
728x90

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

 

1935번: 후위 표기식2

첫째 줄에 피연산자의 개수(1 ≤ N ≤ 26) 가 주어진다. 그리고 둘째 줄에는 후위 표기식이 주어진다. (여기서 피연산자는 A~Z의 영대문자이며, A부터 순서대로 N개의 영대문자만이 사용되며, 길이

www.acmicpc.net

stack 자료구조형을 사용해서 풀었다 

1. 숫자면 stack에 쌓는다 
단 여기서는 숫자대신 ABC... 알파벳 대문자로 표기하고 있는데 이를 ord() 함수를 사용해서 아스키코드 값을 출력해준다 
A가 아스키코드로 65이므로 이를 뺀 값으로 피연산자들을 모은 리스트의 인덱스로 사용했다 
ex) num[ord('A') - 65]

2. 연산자면 pop 두번해서 피연산자들 꺼내고 연산한다

3. 계산 값을 stack 에 다시 쌓는다 
계산할 때 첫번째로 pop한 값이 두번째 피연산자, 두번째로 pop한 값이 첫번째 피연산자가 된다 
마지막으로 stack에 남아 있는 값이 result가 된다 -> 출력값은 소수 2번째자리까지이므로 format() 함수를 사용했다 
"{:.2f}".format(result)

# python에서 A가 아스키코드 65

n = int(input()) # 피연산자 개수
data = input()
num = []
for _ in range(n):
    num.append(int(input()))

stack = []
result = 0

for i in data:
    if i.isalpha():
        stack.append(num[ord(i)-65]) # 숫자면 스택에 쌓는다
    else:
        if i == '*':
            x = stack.pop()
            y = stack.pop()
            stack.append(y*x)
        elif i == '/':
            x = stack.pop()
            y = stack.pop()
            stack.append(y/x)
        elif i == '+':
            x = stack.pop()
            y = stack.pop()
            stack.append(y+x)
        elif i == '-':
            x = stack.pop()
            y = stack.pop()
            stack.append(y-x)
result = stack.pop()

# 소수점 둘째자리까지 출력하기 - 포멧 함수를 사용한다
print("{:.2f}".format(result))

 

728x90