python精選例題筆記（每日一練）——計算後綴表達式的值 Evaluate Postfix String

1.題目

Write a function to evaluate a postfix string.

計算後綴綴表達式 ‘2532*1++3/+’ 的值。

2.解題思路

從左到右掃描表達式：
（1）當前掃描到的是數字，直接入棧；
（2）當前掃描到的是符號，將棧頂兩數字出棧與當前符號計算，將計算後的值入棧；
（3）將表達式全部掃描完畢，棧內剩下的數就是後綴表達式的值。

3.code：

• 小技巧：用如下代碼防止表達式中混有空格。
  infix = ‘’.join([c for c in infix if c != ’ '])

def eval_postfix(postfix: str):
    
    def compute(op1, op2, op):
        if op == '+':
            return op1 + op2
        if op == '-':
            return op1 - op2
        if op == '*':
            return op1 * op2
        if op == '/':
            return op1 / op2
        
    # Remove all white space
    postfix = ''.join([c for c in postfix if c != ' '])
    stk = []
    
    for c in postfix:
        if c.isdigit():
            stk.append(int(c))
        else:
            op2 = stk.pop()
            op1 = stk.pop()
            stk.append(compute(op1, op2, c))
    return stk.pop()


# Call the above function
eval_postfix('2532*1++3/+')

result：

6.0