Algorithm-PS/1918.py at master · lecocococo/Algorithm-PS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import sys
infix = sys.stdin.readline().strip()
op = []

postfix = ''
for ch in infix:

    if ch in "-+*/()":
        if ch == '(':
            op.append(ch)
        elif ch in "*/":
            while op and op[-1] in '*/':
                postfix += op.pop()
            op.append(ch)
        elif ch in '+-':
            while op and op[-1] != '(':
                postfix += op.pop()
            op.append(ch)
        elif ch == ')':
            while op and op[-1] != '(':
                postfix += op.pop()
            op.pop()
    else:
        postfix += ch

while op:
    postfix += op.pop()

print(postfix)