1. 程式人生 > 實用技巧 >LeetCode 150. 逆波蘭表示式求值

LeetCode 150. 逆波蘭表示式求值

150. 逆波蘭表示式求值

Difficulty: 中等

根據,求表示式的值。

有效的運算子包括+,-,*,/。每個運算物件可以是整數,也可以是另一個逆波蘭表示式。

說明:

  • 整數除法只保留整數部分。
  • 給定逆波蘭表示式總是有效的。換句話說,表示式總會得出有效數值且不存在除數為 0 的情況。

示例1:

輸入: ["2", "1", "+", "3", "*"]
輸出: 9
解釋: 該算式轉化為常見的中綴算術表示式為:((2 + 1) * 3) = 9

示例2:

輸入: ["4", "13", "5", "/", "+"]
輸出: 6
解釋: 該算式轉化為常見的中綴算術表示式為:(4 + (13 / 5)) = 6

示例3:

輸入: ["10", "6", "9", "3", "+", "-11", "*", "/", "*", "17", "+", "5", "+"]
輸出: 22
解釋: 
該算式轉化為常見的中綴算術表示式為:
  ((10 * (6 / ((9 + 3) * -11))) + 17) + 5
= ((10 * (6 / (12 * -11))) + 17) + 5
= ((10 * (6 / -132)) + 17) + 5
= ((10 * 0) + 17) + 5
= (0 + 17) + 5
= 17 + 5
= 22

逆波蘭表示式:

逆波蘭表示式是一種字尾表示式,所謂字尾就是指算符寫在後面。

  • 平常使用的算式則是一種中綴表示式,如 ( 1 + 2 ) * ( 3 + 4 )
  • 該算式的逆波蘭表示式寫法為 ( ( 1 2 + ) ( 3 4 + ) * )

逆波蘭表示式主要有以下兩個優點:

  • 去掉括號後表示式無歧義,上式即便寫成 1 2 + 3 4 + * 也可以依據次序計算出正確結果。
  • 適合用棧操作運算:遇到數字則入棧;遇到算符則取出棧頂兩個數字進行計算,並將結果壓入棧中。

Solution

Language: 全部題目

這道題目比較簡單。唯一需要注意的地方是做除法的時候如果除數或被除數中有且僅有一個負數的時候,需要將其轉換為正數之後再做取整的除法。

例如,-6 // 132,向下取整的結果是-1,而我們期望的結果是0,所以合理的計算方式為-1 * (6 // 132) = 0

class Solution:
  def evalRPN(self, tokens: List[str]) -> int:
    stack = []
    for _, e in enumerate(tokens):
      if e not in ['+', '-', '*', '/']:
        stack.append(int(e))
      else:
        i = 0
        tmp = []
        while i < 2:
          tmp.append(stack.pop())
          i += 1
        if e == '+':
          stack.append(tmp[1] + tmp[0])
        elif e == '-':
          stack.append(tmp[1] - tmp[0])
        elif e == '/':
          t = abs(tmp[1]) // abs(tmp[0])
          if tmp[1] * tmp[0] < 0:
            stack.append(-1 * t)
          else:
            stack.append(t)
        else:
          stack.append(tmp[1] * tmp[0])
    return stack[0]