(cache)[Wandbox]三へ( へ՞ਊ ՞)へﾊｯﾊｯ

Python

Load template

Runtime options:

🐕 Corporate Sponsors:

🐶 Personal Sponsors:

About sponsors(Japanese)

created by

anonymous

1 minute ago

​x
                    converted_expr.append(calc(op_stack.pop()[1]))
 
import sys
​
def add(a, b):
    return a + b
​
def mul(a, b):
    return a * b
​
def add_num(n):
    return lambda s: s.append(n)
​
def calc(f):
    def do_calc(s):
        s[-2] = f(s[-2], s[-1])
        s.pop()
    return do_calc
​
q = sys.stdin.readline().rstrip()
​
converted_expr = []
op_stack = []
​
zero = ord('0')
​
current = -1
​
prev_kokka = False
​
for c in q:
    v = ord(c) - zero
    if c == '+':
        if current >= 0:
            converted_expr.append(add_num(current))
            current = -1
        while len(op_stack) > 0 and op_stack[-1][0] >= 0:
            converted_expr.append(calc(op_stack.pop()[1]))
        op_stack.append((0, add))
    elif c == '*':
        if current >= 0:
            converted_expr.append(add_num(current))
            current = -1
        while len(op_stack) > 0 and op_stack[-1][0] >= 1:
            converted_expr.append(calc(op_stack.pop()[1]))
        op_stack.append((1, mul))
    elif c == '(':
        if current >= 0:
            converted_expr.append(add_num(current))
            current = -1
            while len(op_stack) > 0 and op_stack[-1][0] >= 1:
                converted_expr.append(calc(op_stack.pop()[1]))
            op_stack.append((1, mul))
        op_stack.append((-1, None))
    elif c == ')':
        if current >= 0:
            converted_expr.append(add_num(current))
            current = -1
        while len(op_stack) > 0 and op_stack[-1][0] >= 0:
            converted_expr.append(calc(op_stack.pop()[1]))
        op_stack.pop()
    elif 0 <= v and v <= 9:
        if current < 0:
            current = 0

Editor settings:

Smart Indent

$ python3 prog.py

Stdin

​
 
3+4(2*6+3)​

Run (or Ctrl+Enter)

#permlink

Code

Start

Finish

No Wrap