各位题友大家好! 今天是 @负雪明烛 坚持日更的第 46 天。今天力扣上的每日一题是「227. 基本计算器 II」。
解题思路
重点:
- 有 +
, -
,*
,/
四种运算符;
- 没有括号。
既然包含 4 种运算符,那么就有运算符优先级的问题:先 *
,/
,再 +
, -
。本题解的思路就是把所有的 *
,/
先计算出来,最后计算只有 +
, -
运算符的表达式。
使用一个「栈」只保存需要进行
+
,-
运算符的所有数字(把-
运算符改成负数放到栈中)。
- 如果遇到*
,/
运算,则需要把结果先计算出来,也放到栈里;在把所有乘除法计算完成之后,最后对栈内数字求和。
但是注意栈顶元素由于后面可能遇到 *
,/
运算符,所以栈顶元素有可能还会被弹出来,跟后面的运算符做计算。
思路是不是很简单?
具体实现的时候,按照下面的思路来考虑。
一个运算符表达式分为三个部分,可以用下面的情况表示:
数字① 运算符② 数字③
- 数字①,在栈中保存,为栈顶的元素;
- 运算符②,用一个变量 pre_op 保存;
- 数字③,用一个变量 num 保存。
操作情况:
运算符②,决定了现在的操作:
- 如果 运算符② 为
+
,-
:如果是+,
是把数字③入栈;如果 -,
是把**-数字③**
入栈。 - 如果 运算符② 为
*
,/
,则需要计算 `数字① 运算符② 数字③`,然后把结果入栈。
*
这样遍历一次后,优先把所有的 `,
/` 都计算出来,而且与需要做加减运算的数字一起,全都都放到了栈中,对栈求和,即为最终的结果。
代码
我是用 python3 写的代码,这里会有个坑,坑爹的Python3「地板除 //
」。比如在 Python3 中 (-3) // 2 = -2
的,和 c++ 计算负数除法的结果不一样。因此如果在计算的过程中,如果遇到负数的除法,先使用的用「浮点除 /
」再取整的方式获得和 c++一样的结果。
代码中,当 字符串遍历结束 或者 当前字符each 为 '+-*/'
的时候,再进行上面说的运算的原因是,此时数字③已经结束。
class Solution:
def calculate(self, s):
stack = []
pre_op = '+'
num = 0
for i, each in enumerate(s):
if each.isdigit():
num = 10 * num + int(each)
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 = each
num = 0
return sum(stack)
- 时间复杂度:$O(N)$
- 空间复杂度:$O(N)$
刷题心得
今天的题目思路还是比较简单的,细节略多,注意保证细节的正确性。
OK,以上就是 @负雪明烛 写的今天题解的全部内容了,如果你觉得有帮助的话,求赞、求关注、求收藏。如果有疑问的话,请在下面评论,我会及时解答。
关注我,你将不会错过我的精彩动画题解、面试题分享、组队刷题活动,进入主页 @负雪明烛 右侧有刷题组织,从此刷题不再孤单。
祝大家牛年大吉!AC 多多,Offer 多多!我们明天再见!