227.基本计算器Ⅱ

题目:
给你一个字符串表达式 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)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值