力扣刷题笔记:227.基本计算器 II(栈来处理运算符的优先级,主要是逻辑思路要清晰,代码很简单,小表弟都看懂了)

题目:

227、基本计算器 II

给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。

整数除法仅保留整数部分。

示例 1:

输入:s = “3+2*2”
输出:7

示例 2:

输入:s = " 3/2 "
输出:1

示例 3:

输入:s = " 3+5 / 2 "
输出:5

提示:

1 <= s.length <= 3 * 10^5
s 由整数和算符 (’+’, ‘-’, ‘*’, ‘/’) 组成,中间由一些空格隔开
s 表示一个 有效表达式
表达式中的所有整数都是非负整数,且在范围 [0, 2^31 - 1] 内
题目数据保证答案是一个 32-bit 整数

题解思路:

由于乘除优先于加减计算,因此不妨考虑先进行所有乘除运算,并将这些乘除运算后的整数值放回原表达式的相应位置,则随后整个表达式的值,就等于一系列整数加减后的值。

基于此,我们可以用一个栈,保存这些(进行乘除运算后的)整数的值。对于加减号后的数字,将其直接压入栈中;对于乘除号后的数字,可以直接与栈顶元素计算,并替换栈顶元素为计算后的结果。

具体来说,遍历字符串 s,并用变量 preSign 记录每个数字之前的运算符,对于第一个数字,其之前的运算符视为加号。每次遍历到数字末尾时,根据 preSign 来决定计算方式:

加号:将数字压入栈;
减号:将数字的相反数压入栈;
乘除号:计算数字与栈顶元素,并将栈顶元素替换为计算结果。
代码实现中,若读到一个运算符,或者遍历到字符串末尾,即认为是遍历到了数字末尾。处理完该数字后,更新 preSign 为当前遍历的字符。

遍历完字符串 s 后,将栈中元素累加,即为该字符串表达式的值。

题解python代码:

class Solution:
    def calculate(self, s: str) -> int:
        stack = []
        num, preSign = 0, ""   # preSign为当前运算符的上一个符号
        for i in range(len(s)):
            if s[i].isdigit():
                num = num*10+int(s[i])
            if not s[i].isdigit() and s[i]!=" " or i==len(s)-1:  # 在下一个符号或字符结束处进行计算
                if preSign=="":
                    stack.append(num)
                elif preSign=="+" or preSign=="-":
                    stack.append(num if preSign=="+" else -num)
                elif preSign=="*":
                    stack[-1] *= num
                elif preSign=="/":
                    stack[-1] /= num
                    stack[-1] = int(stack[-1])   # 取整
                num = 0
                preSign = s[i]
            # print(s[i], preSign, stack)
        return sum(stack)

作者:a-qing-ge
链接:https://leetcode-cn.com/problems/basic-calculator-ii/solution/guan-fang-ti-jie-zhan-pythonban-by-a-qin-p00z/
来源:力扣(LeetCode)https://leetcode-cn.com/problems/basic-calculator-ii/

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值