题目:
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
整数除法仅保留整数部分。
示例 1:
输入:s = “3+2*2”
输出:7
示例 2:
输入:s = " 3/2 "
输出:1
示例 3:
输入:s = " 3+5 / 2 "
输出:5
提示:
1 <= s.length <= 3 * 105
s 由整数和算符 (’+’, ‘-’, ‘*’, ‘/’) 组成,中间由一些空格隔开
s 表示一个 有效表达式
表达式中的所有整数都是非负整数,且在范围 [0, 231 - 1] 内
题目数据保证答案是一个 32-bit 整数
题目链接:
https://leetcode-cn.com/problems/basic-calculator-ii
解题思路
重点:
有 +, - ,*,/ 四种运算符;
没有括号。
既然包含 4 种运算符,那么就有运算符优先级的问题:先 *,/ ,再 +, - 。本题解的思路就是把所有的 *,/ 先计算出来,最后计算只有 +, - 运算符的表达式。
使用一个「栈」只保存需要进行 +, - 运算符的所有数字(把- 运算符改成负数放到栈中)。
如果遇到 *,/ 运算,则需要把结果先计算出来,也放到栈里;
在把所有乘除法计算完成之后,最后对栈内数字求和。
但是注意栈顶元素由于后面可能遇到 *,/ 运算符,所以栈顶元素有可能还会被弹出来,跟后面的运算符做计算。
具体实现的时候,按照下面的思路来考虑。
一个运算符表达式分为三个部分,可以用下面的情况表示:
数字①, 运算符②, 数字③
数字①,运算符②,数字③
数字①,在栈中保存,为栈顶的元素;
运算符②,用一个变量 pre_op 保存;
数字③,用一个变量 num 保存。
操作情况:
运算符②,决定了现在的操作:
如果 运算符② 为 +, - :如果是+,是把数字③入栈;如果 -,是把 数字③取反 入栈。
如果 运算符② 为 *,/ ,则需要计算 数字① 运算符② 数字③,然后把结果 入栈。
这样遍历一次后,优先把所有的 *,/ 都计算出来,而且与需要做加减运算的数字一起,全都都放到了栈中,对栈求和,即为最终的结果。
作者:fuxuemingzhu
链接:https://leetcode-cn.com/problems/basic-calculator-ii/solution/xian-cheng-chu-zai-jia-jian-yong-zhan-ba-hplr/
代码:
class Solution:
def calculate(self, s):
stack = []
pre_op = '+'
num = 0
for i, v in enumerate(s):
if each.isdigit():
num = 10 * num + int(v)
if i == len(s) - 1 or each in '+-*/':
if pre_op == '+':
stack.append(num)
elif pre_op == '-':
stack.append(-num)
elif pre_op == '*':
stack.append(stack.pop() * num)
elif pre_op == '/':
top = stack.pop()
if top < 0:
stack.append(int(top / num))
else:
stack.append(top // num)
pre_op = v
num = 0
return sum(stack)