[BOJ] 1918 | 후위 표기식

1918 | 후위 표기식
https://www.acmicpc.net/problem/1918

 

 

 

작성 코드

 

express = input()

stack = []
answer = []

for exp in express:

    if exp not in ['+', '-', '*', '/', '(', ')']:
        answer.append(exp)
    
    elif exp == '(':
        stack.append(exp)

    elif exp == ')':
        while stack[-1] != '(':
            answer += stack.pop()
        stack.pop()
    
    elif exp in ['+', '-']:
        while stack and (stack[-1] in ['*', '/', '-', '+']):
            answer += stack.pop()
        stack.append(exp)
    
    elif exp in ['*', '/']:
        while stack and (stack[-1] in ['/', '*']):
            answer += stack.pop()
        stack.append(exp)


answer += stack[::-1]

print(''.join(answer))

 

• 일일이 우선 순위 비교

 

 

이상 코드

 

s = input()
priority = {'*': 2, '/': 2, '+': 1, '-': 1, '(': 0}
op = []

for c in s:
    if 'A' <= c <= 'Z':
        print(c, end="")
    elif c == '(':
        op.append(c)
    elif c == ')':
        while op and op[-1] != '(':
            print(op.pop(),end="")
        op.pop()
    else:
        while op and priority[c] <= priority[op[-1]]:
            print(op.pop(), end="")
        op.append(c)
        
while op:
    print(op.pop(), end="")

 

• 알파벳(대문자)인지 확인할 때  'A' <= c <= 'Z'             
• 딕셔너리에 우선순위 저장 후 비교