今天学逆波兰表达式撰写还顺便复习了一下二叉树的前中后序遍历:
前序遍历:根左右
中序遍历:左根右
后序遍历:左右根
本题两个要点:
a.判断字符串中的元素是数字还是字符串
因为Python 中没有一个函数可以判断一个字符串是否为合理的整数(包括正、负数)。str.isdigit() 可以判断正数,但是无法判断负数。
解决方法:
使用 int() 函数,并做 try-except 。
如果是整数,那么可以用 int() 转成数字;
如果是运算符,那么 int() 会报错,从而进入 except 中。
b.python 的整数除法是向下取整,而不是向零取整。
python2 的除法 "/" 是整数除法, "-3 / 2 = -2" ;
python3 的地板除 "//" 是整数除法, "-3 // 2 = -2" ;
python3 的除法 "/" 是浮点除法, "-3 / 2 = -1.5" ;
而 C++/Java 中的整数除法是向零取整。C++/Java 中 "-3 / 2 = -1" .本题的题意(一般情况)都是要求向零取整的。
解决方法:
对 Python 的整数除法问题,可以用 int(num1 / float(num2)) 来做,即先用浮点数除法,然后取整。
无论如何,浮点数除法都会得到一个浮点数,比如 "-3 / 2.0 = 1.5" ;
此时再取整,就会得到整数部分,即 float(-1.5) = -1 。
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
stack = []
for token in tokens:
try:
stack.append(int(token))
except:
num2 = stack.pop()
num1 = stack.pop()
stack.append(self.evaluate(num1,num2, token))
return stack[0]
def evaluate(self,num1,num2,op):
if op == "+":
return num1+num2
if op == "-":
return num1-num2
if op == "*":
return num1 *num2
if op == "/":
return int(num1/float(num2))