227. 基本计算器 II Python

文章目录


一、题目描述

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

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

你可以假设给定的表达式总是有效的。所有中间结果将在 [-2^31, 2^31 - 1]的范围内。

注意 :不允许使用任何将字符串作为数学表达式计算的内置函数,比如 eval() 。

示例 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 整数

二、代码

代码如下:

c 复制代码
class Solution:
    def calculate(self, s: str) -> int:
        def is_operator(char):
            return char in "+-*/^"

        def infix_to_postfix(expression):
            def precedence(operator):
                precedence_dict = {'+': 1, '-': 1, '*': 2, '/': 2, '^': 3}
                return precedence_dict.get(operator, 0)

            def infix_to_postfix_internal(expression_tokens):
                output = []
                operator_stack = []

                for token in expression_tokens:
                    if token.isnumeric():  # 如果是数字,直接添加到输出
                        output.append(token)
                    elif token == '(':  # 如果是左括号,压入栈
                        operator_stack.append(token)
                    elif token == ')':  # 如果是右括号,将栈顶的操作符弹出并添加到输出,直到遇到左括号
                        while operator_stack and operator_stack[-1] != '(':
                            output.append(operator_stack.pop())
                        if operator_stack and operator_stack[-1] == '(':
                            operator_stack.pop()
                    elif is_operator(token):  # 如果是操作符,处理操作符的优先级
                        while (operator_stack and
                            operator_stack[-1] != '(' and
                            precedence(token) <= precedence(operator_stack[-1])):
                            output.append(operator_stack.pop())
                        operator_stack.append(token)

                while operator_stack:  # 将栈中剩余的操作符全部添加到输出
                    output.append(operator_stack.pop())

                return ' '.join(output)

            # 将输入的表达式字符串按空格分割成标记列表
            expression_tokens = expression.split()
            # 调用内部函数进行转换
            postfix_expression = infix_to_postfix_internal(expression_tokens)
            return postfix_expression

        def add_spaces_to_infix(expression):
            operators = "+-*/^"
            result = []

            for char in expression:
                if char in operators or char in "()":
                    result.append(f" {char} ")
                else:
                    result.append(char)

            return ''.join(result)

        infix_expression = s.replace(" ", "")
        print(infix_expression)
        spaced_infix_expression = add_spaces_to_infix(infix_expression)
        print(spaced_infix_expression)
        postfix_expression = infix_to_postfix(spaced_infix_expression)
        print("中缀表达式:", spaced_infix_expression)
        print("后缀表达式:", postfix_expression)

        stack = []

        for token in postfix_expression.split():
            if token.isnumeric():
                stack.append(int(token))
            elif is_operator(token):
                operand2 = stack.pop()
                operand1 = stack.pop()
                if token == '+':
                    result = operand1 + operand2
                elif token == '-':
                    result = operand1 - operand2
                elif token == '*':
                    result = operand1 * operand2
                elif token == '/':
                    if operand2 == 0:
                        raise ValueError("Division by zero")
                    result = int(operand1 / operand2)
                stack.append(result)

        print(int(stack[0]))
        return int(stack[0])

三、解题思路

本题本质是要求通过字符串来计算表达式,且不能直接使用eval方法对字符串直接进行转化计算。本题解题思路为:将字符串(前缀表达式)转化为后缀表达式,然后通过计算后缀表达式得到结果。

① 因为涉及到字符可能会出现2位数及其以上的情况,如果之间转化为后缀表达式,则会导致数字连接到一块,所以需要对数字和运算符进行分隔,将表达式转化为如下格式:"12+4/5" => "12 + 4 / 5"

② 转化为后缀表达式,当遇到2位数及以上的数字时,需要将其看做是一个数,用空格分隔不同数。例如:

中缀表达式: 12 + 4 / 5

后缀表达式: 12 4 5 / +

不同的数之间用空格分开

③ 计算后缀表达式,找第一个运算符,向左找最近的2个数进行计算,重复这一过程,最后得到一个值返回即可。

相关推荐
倒头就睡的小比特2 天前
算法竞赛C++常用的STL
c++·算法
默_笙2 天前
🍙 给每个请求过安检:FastAPI 是怎么把校验写进类型注解的
python
小羊没烦恼!2 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
qq_426003962 天前
启动playwright录制codegen生成自动化测试脚本
python·自动化
虎头金猫2 天前
4K 视频总卡在公网带宽?用 N1 + OpenList 把网盘播放链路重新理顺
运维·服务器·网络·python·容器·beautifulsoup·pandas
长沙三为智能科技2 天前
家政小程序开发从0到上线:五阶段交付流程与验收清单
python
伞伞悦读2 天前
【第38期】Python 模块与包详解:import、from、模块搜索路径、包结构和 __init__
开发语言·python
猎头南楼2 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
只睡四小时2 天前
Canvas 弹道联机实战:700 行 + 固定时间步长
python·websocket·html5·游戏开发·canvas
C语言小火车2 天前
C/C++ 为什么需要编译器?
开发语言·c++