LeetCode 150:逆波兰表达式求值
往期回顾
今天来做逆波兰表达式求值,可能臭宝们一看“逆波兰表达式”这个洋气的名字,下意识的会觉得难搞。
其实这个词只是看着唬人,就是纸老虎。
逆波兰表达式又叫后缀表达式,是将运算符写在操作数之后。
像我们平时写的是类似 a + b 这种表达式,这叫中缀表达式,符合我们作为人的思维方式,而后波兰表达式则是写成 ab+。
简单说下为啥要弄个逆波兰表达式这个玩意。
原因就是从我们人的角度来看,简单的一批的中缀表达式,在计算机看来,简直是复杂的无与伦比,因为计算机普遍采用的是栈式结构,执行的是先入后出的操作,所以逆波兰式在它眼里才是个正常的娃儿。
LeetCode 150:逆波兰表达式求值
题意
根据逆波兰表示法,求表达式的值。
有效算符包括 + - * /。运算对象可以是整数,可以是其他逆波兰表达式。
示例
输入:tokens = ["2", "1", "+", "3", "*"]
输出:9
解释:该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9
输入:tokens = ["4","13","5","/","+"]
输出:6
解释:该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6
提示
整数除法只保留整数部分。
给定逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况。
1 <= tokens.length <= 104
tokens[i] 要么是一个算符("+"、"-"、"*" 或 "/"),要么是一个在范围 [-200, 200] 内的整数。
题目解析
水题,难度中等,感觉难度都堆在“逆波兰表达式”这几个字上。
逆波兰表达式主要有以下两个优点:
去掉括号后表达式无歧义,上式即便写成 1 2 + 3 4 + * 也可以依据次序计算出正确结果。
适合用栈操作运算:遇到数字则入栈;遇到算符则取出栈顶两个数字进行计算,并将结果压入栈中。
在前面说了,逆波兰表达式为计算机这种思维而生,那这个逆波兰表达式求值显然也是用栈来解决。
逆波兰表达式严格遵守从左到右的运算,所以解法也很简单,其实就是维护一个栈,从左到右扫描 tokens:
如果遇到数,数入栈。
如果遇到操作符,栈顶两个元素出栈,先出栈的数在运算符右侧,后出栈的在运算符左侧,运算完后,将运算值入栈。
等全部扫描完以后,栈内剩下的那个数,就是最后的结果。
图解
假设 tokens = ["2", "1", "+", "3", "*"]。
首先维护一个栈,根据“题目解析”中说的,碰到数先入栈,那第 1 步和第 2 步,数字 2 和数字 1 入栈。
# 初始化栈
stack = []
for c in tokens:
# 如果当前字符不是操作符,入栈
if c not in ['+', '-', '*', '/']:
stack.append(int(c))
接下来第 3 步,当前元素为操作符,所以栈顶两个元素出栈,先出栈的在操作符右侧,后出栈的在操作符左侧,运算完,新值入栈。
# 先出栈的数在操作符右侧
right = stack.pop()
# 后出栈的数在操作符左侧
left = stack.pop()
# 进行相应运算,运算完的数值入栈
if c == '+':
stack.append(left + right)
往下第 4 步,是数字,入栈。
最后第 5 步,碰到操作符,出栈,相乘,入栈。
# 进行相应运算,运算完的数值入栈
if c == '+':
stack.append(left + right)
elif c == '-':
stack.append(left - right)
elif c == '*':
stack.append(left * right)
else:
# 除法这注意一下
stack.append(int(left / right))
至此,结束,返回栈中的值即可。
# 最后只剩下一个数值,就是最终结果
return stack[0]
因为这个只需要从左到右遍历 tokens 数组一次,所以时间复杂度为 O(N)。
做的过程中维护了一个栈存储计算过程中的数,所以空间复杂度为 O(N)。
代码实现
Python 代码实现
class Solution:
def evalRPN(self, tokens: List[str]) -> int:
# 初始化栈
stack = []
for c in tokens:
# 如果当前字符不是操作符,入栈
if c not in ['+', '-', '*', '/']:
stack.append(int(c))
# 如果当前字符是操作符
else:
# 先出栈的数在操作符右侧
right = stack.pop()
# 后出栈的数在操作符左侧
left = stack.pop()
# 进行相应运算,运算完的数值入栈
if c == '+':
stack.append(left + right)
elif c == '-':
stack.append(left - right)
elif c == '*':
stack.append(left * right)
else:
# 除法这注意一下
stack.append(int(left / right))
# 最后只剩下一个数值,就是最终结果
return stack[0]
Java 代码实现
class Solution {
public int evalRPN(String[] tokens) {
Deque
stack = new LinkedList (); int n = tokens.length;
for (int i = 0; i < n; i++) {
String token = tokens[i];
if (isNumber(token)) {
stack.push(Integer.parseInt(token));
} else {
int num2 = stack.pop();
int num1 = stack.pop();
switch (token) {
case "+":
stack.push(num1 + num2);
break;
case "-":
stack.push(num1 - num2);
break;
case "*":
stack.push(num1 * num2);
break;
case "/":
stack.push(num1 / num2);
break;
default:
}
}
}
return stack.pop();
}
好啦,图解逆波兰表达式的值到这就结束啦。
这道题就是多学了一个概念,解题方法和代码没什么难度。
最后也欢迎大家加入帅地的知识星球。帅地会在星球知无不言,无论是 offer 选择,职业规划还是学习路线,帅地都会在 48 小时以内答复你的问题,并且根据你自身的情况,为你量身定制学习路线。
你有任何的疑惑,帅地都会答复你,并且星球里也有一群和你相同年龄的小伙伴在奋斗,都是未来的卷王
星球还提供了超全学习攻略,小白只需要跟着我提供的攻略学习就可以了
帅地会在星球会提供如下服务
1、【一对一咨询服务】48 小时以内超详细回答你的任何问题,包括写作等等,这是知识星球最重要的功能。最近一周就回答了十几个 offer 选择,校招等学习路线问题。
2、【学习攻略】校招这方面比较有经验,帅地会提供完整的学习攻略,小白跟着帅地说的学习就行,。
3、简历修改,项目等学习资源,offer 收割机嘉宾分享等等。
目前星球是专注于校招,在校生学习指导这块,一定可以让你少走弯路,已经有 590+ 位小伙伴加入,这里还有一些 20 元的优惠卷,如果你信的过帅地,那么欢迎你的加入。
点击阅读原文可以了解详情