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

解题思路

重点:
- +-*/ 四种运算符;
- 没有括号。

既然包含 4 种运算符,那么就有运算符优先级的问题:先 */ ,再 +-本题解的思路就是把所有的 */ 先计算出来,最后计算只有 +- 运算符的表达式。

  • 使用一个「栈」只保存需要进行 +- 运算符的所有数字(把- 运算符改成负数放到栈中)。
    - 如果遇到 */ 运算,则需要把结果先计算出来,也放到栈里;

  • 在把所有乘除法计算完成之后,最后对栈内数字求和

但是注意栈顶元素由于后面可能遇到 */ 运算符,所以栈顶元素有可能还会被弹出来,跟后面的运算符做计算。

思路是不是很简单?

具体实现的时候,按照下面的思路来考虑。

一个运算符表达式分为三个部分,可以用下面的情况表示:

数字① 运算符② 数字③

  • 数字①,在栈中保存,为栈顶的元素;
    - 运算符②,用一个变量 pre_op 保存;
    - 数字③,用一个变量 num 保存。

操作情况:
运算符②,决定了现在的操作:

  • 如果 运算符② 为 +- :如果是+,是把数字③入栈;如果 -是把 **-数字③** 入栈。
  • 如果 运算符② 为 */ ,则需要计算 `数字① 运算符② 数字③`,然后把结果入栈。

*
这样遍历一次后,优先把所有的 `
/` 都计算出来,而且与需要做加减运算的数字一起,全都都放到了栈中,对栈求和,即为最终的结果。

代码

我是用 python3 写的代码,这里会有个坑,坑爹的Python3「地板除 // 」。比如在 Python3 中 (-3) // 2 = -2 的,和 c++ 计算负数除法的结果不一样。因此如果在计算的过程中,如果遇到负数的除法,先使用的用「浮点除 / 」再取整的方式获得和 c++一样的结果。

代码中,当 字符串遍历结束 或者 当前字符each 为 '+-*/' 的时候,再进行上面说的运算的原因是,此时数字③已经结束。

  1. class Solution:
  2. def calculate(self, s):
  3. stack = []
  4. pre_op = '+'
  5. num = 0
  6. for i, each in enumerate(s):
  7. if each.isdigit():
  8. num = 10 * num + int(each)
  9. if i == len(s) - 1 or each in '+-*/':
  10. if pre_op == '+':
  11. stack.append(num)
  12. elif pre_op == '-':
  13. stack.append(-num)
  14. elif pre_op == '*':
  15. stack.append(stack.pop() * num)
  16. elif pre_op == '/':
  17. top = stack.pop()
  18. if top < 0:
  19. stack.append(int(top / num))
  20. else:
  21. stack.append(top // num)
  22. pre_op = each
  23. num = 0
  24. return sum(stack)
  • 时间复杂度:$O(N)$
  • 空间复杂度:$O(N)$

刷题心得

今天的题目思路还是比较简单的,细节略多,注意保证细节的正确性。


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

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

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