各位题友大家好! 今天是 @负雪明烛 坚持日更的第 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 = 0for 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 = eachnum = 0return sum(stack)
- 时间复杂度:$O(N)$
- 空间复杂度:$O(N)$
刷题心得
今天的题目思路还是比较简单的,细节略多,注意保证细节的正确性。
OK,以上就是 @负雪明烛 写的今天题解的全部内容了,如果你觉得有帮助的话,求赞、求关注、求收藏。如果有疑问的话,请在下面评论,我会及时解答。
关注我,你将不会错过我的精彩动画题解、面试题分享、组队刷题活动,进入主页 @负雪明烛 右侧有刷题组织,从此刷题不再孤单。
祝大家牛年大吉!AC 多多,Offer 多多!我们明天再见!
