各位题友大家好! 今天是 @负雪明烛 坚持日更的第 45 天。今天力扣上的每日一题是「224. 基本计算器」。

解题思路

重点:
- 本题目只有 "+", "-" 运算,没有 "*" , "/" 运算,因此少了不同运算符优先级的比较;
- 遇到小括号,应该先算括号里面的表达式;

递归

一个表达式分为三部分:

左边表达式① 运算符③ 右边表达式②

本题中,左边和右边的表达式可以是一个数字,也可以是一个括号包起来的表达式;运算符可以是加/减。

小学数学告诉我们,一个只包含加减和括号的表达式,我们可以从左到右计算,遇到括号就先算括号里面的。具体来说就是先计算左边的表达式,再计算右边表达式,最后根据运算符,计算 ①和②的运算

用题目实例 "(1+(4+5+2)-3)+(6+8)" 来说明运算符计算的顺序:

根据上面的分析可知,当我们在计算一个表达式的时候,需要先计算左边表达式①,然后需要把①的结果和运算符③保存起来,再需要计算右边表达式②,最后计算①和②的运算。这个操作就是递归!!

递归的程序可以用「栈」来模拟:栈为了保存左边表达式①的计算结果和运算符③,在计算右边表达式③的结果之后,从栈中取出运算符③和①的结果,再进行计算整个表达式的结果。

肯定有朋友想问了,用栈保存左边表达式结果的话,当遇到嵌套的括号怎么办?比如 (1 + (2 + (3 + 4)))。答案是:栈顶保留的是最里层嵌套的运算,弹出栈的时候,正好先算的是最里面括号的,再算外边括号的。这种情况时,栈里面保存的是 [“1”, “+”, “2”, “+”, “3”, “+”],然后遇到 4,此时计算的是 3 + 4,然后算 7 + 2,再算 9 + 1。这类似于递归

代码

代码里面:

  • res 表示左边表达式除去栈内保存元素的计算结果;
  • sign 表示运算符;
  • num 表示当前遇到的数字,会更新到 res 中;
  • 用栈保存遇到左括号时前面计算好了的结果和运算符。

操作的步骤是:

  • 如果当前是数字,那么更新计算当前数字;
    - 如果当前是操作符+或者-,那么需要更新计算当前计算的结果res,并把当前数字num设为0,sign设为正负,重新开始;
    - 如果当前是 ( ,那么说明遇到了右边的表达式,而后面的小括号里的内容需要优先计算,所以要把res,sign进栈,更新res和sign为新的开始;
    - 如果当前是 ) ,那么说明右边的表达式结束,即当前括号里的内容已经计算完毕,所以要把之前的结果出栈,然后计算整个式子的结果;
    - 最后,当所有数字结束的时候,需要把最后的一个 num 也更新到 res 中。
  1. class Solution(object):
  2. def calculate(self, s):
  3. res, num, sign = 0, 0, 1
  4. stack = []
  5. for c in s:
  6. if c.isdigit():
  7. num = 10 * num + int(c)
  8. elif c == "+" or c == "-":
  9. res += sign * num
  10. num = 0
  11. sign = 1 if c == "+" else -1
  12. elif c == "(":
  13. stack.append(res)
  14. stack.append(sign)
  15. res = 0
  16. sign = 1
  17. elif c == ")":
  18. res += sign * num
  19. num = 0
  20. res *= stack.pop()
  21. res += stack.pop()
  22. res += sign * num
  23. return res
  • 时间复杂度:$O(N)$
  • 空间复杂度:$O(N)$

刷题心得

今天的题目其实不难,主要是思路要理清楚。

参考资料:

  1. GrandYang
  2. 负雪明烛

OK,以上就是 @负雪明烛 写的今天题解的全部内容了,如果你觉得有帮助的话,求赞、求关注、求收藏。如果有疑问的话,请在下面评论,我会及时解答。

关注我,你将不会错过我的精彩动画题解、面试题分享、组队刷题活动,进入主页 @负雪明烛 右侧有刷题组织,从此刷题不再孤单。

祝大家牛年大吉!AC 多多,Offer 多多!我们明天再见!