跳至主要內容

36. 后缀表达式


36. 后缀表达式open in new window

🟠   🔖  数组 数学  🔗 力扣open in new window

题目

根据 逆波兰表示法open in new window,求该后缀表达式的计算结果。

有效的算符包括 +-*/ 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。

说明:

  • 整数除法只保留整数部分。
  • 给定逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况。

示例 1:

输入: tokens = ["2","1","+","3","*"]

输出: 9

解释: 该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9

示例 2:

输入: tokens = ["4","13","5","/","+"]

输出: 6

解释: 该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6

示例 3:

输入: tokens = ["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 <= tokens.length <= 10^4
  • tokens[i] 要么是一个算符("+""-""*""/"),要么是一个在范围 [-200, 200] 内的整数

逆波兰表达式:

逆波兰表达式是一种后缀表达式,所谓后缀就是指算符写在后面。

  • 平常使用的算式则是一种中缀表达式,如 ( 1 + 2 ) * ( 3 + 4 )
  • 该算式的逆波兰表达式写法为 ( ( 1 2 + ) ( 3 4 + ) * )

逆波兰表达式主要有以下两个优点:

  • 去掉括号后表达式无歧义,上式即便写成 1 2 + 3 4 + * 也可以依据次序计算出正确结果。
  • 适合用栈操作运算:遇到数字则入栈;遇到算符则取出栈顶两个数字进行计算,并将结果压入栈中。

注意

本题与 LeetCode 第 150 题 相同。

解题思路

复杂度分析

逆波兰表达式发明出来就是为了方便计算机运用「栈」进行表达式运算的,其运算规则如下:

按顺序遍历逆波兰表达式中的字符,如果是数字,则放入栈;如果是运算符,则将栈顶的两个元素拿出来进行运算,再将结果放入栈。对于减法和除法,运算顺序别搞反了,栈顶第二个数是被除(减)数。

  1. 遍历输入

    • 遍历输入的字符串数组 tokens,对于每个元素:
      • 操作数:如果当前元素是数字(字符串形式),将其转换为数字并推入栈中。
      • 操作符:如果当前元素是操作符(+, -, *, /),需要执行相应的操作:
        • 从栈中弹出最近的两个操作数。
        • 根据操作符计算结果并将其压入栈中。
  2. 实现操作

    • 对于每个操作符,进行相应的计算:
      • 加法(+:弹出两个数字相加,将结果压入栈中。
      • 减法(-:弹出两个数字,计算第二个减第一个的结果,然后压入栈中。
      • 乘法(*:弹出两个数字相乘,结果压入栈中。
      • 除法(/:注意要处理整数除法的取整方式(向零取整),将第二个数字除以第一个数字的结果压入栈中。
  3. 最终结果

    • 遍历结束后,栈中只会剩下一个数字,即为最终结果,返回这个值。

复杂度分析

  • 时间复杂度O(n),其中 ntokens 数组的长度,只需遍历整个数组一次,每个操作(如加法、减法、乘法、除法和入栈、出栈操作)都是常数时间操作。

  • 空间复杂度O(n),在最坏情况下,栈可能会存储所有的操作数,例如在输入全为数字的情况下,栈的最大大小为 O(n)

代码

/**
 * @param {string[]} tokens
 * @return {number}
 */
var evalRPN = function (tokens) {
	let stack = [];
	for (let i of tokens) {
		if (i == '+') {
			stack.push(stack.pop() + stack.pop());
		} else if (i == '-') {
			stack.push(-1 * stack.pop() + stack.pop());
		} else if (i == '*') {
			stack.push(stack.pop() * stack.pop());
		} else if (i == '/') {
			const temp = stack.pop();
			stack.push((stack.pop() / temp) | 0);
		} else {
			stack.push(Number(i));
		}
	}
	return stack.pop();
};