四则运算解析器 | 豆包MarsCode AI刷题

110 阅读4分钟

问题描述

四则运算解析器是一个基础的编程问题,它要求我们实现一个能够解析并计算包含加、减、乘、除四种运算的数学表达式的程序。这个问题在计算机科学中非常常见,尤其是在编译原理和编程语言设计中。实现一个四则运算解析器可以帮助我们理解如何解析和执行程序代码。

问题背景

在编程语言中,四则运算是最基本的功能之一。无论是高级语言如Python、Java,还是低级语言如C或C++,都内置了对四则运算的支持。然而,这些运算是如何被解析和执行的,对于初学者来说可能并不清晰。通过实现一个简单的四则运算解析器,我们可以深入了解这一过程。

解题思路

实现四则运算解析器可以通过多种方法,包括递归下降解析、栈等。这里我们采用一种相对简单的方法,即利用两个栈(一个用于存储数字,一个用于存储运算符)来实现。

  1. 初始化两个栈:一个用于存储数字(数字栈),另一个用于存储运算符(运算符栈)。

  2. 遍历表达式:从左到右遍历表达式的每个字符。

  3. 处理数字:如果当前字符是数字,则将其读入一个临时变量中,直到遇到非数字字符。

  4. 处理运算符:如果当前字符是运算符,比较其与运算符栈顶元素的优先级:

    • 如果当前运算符优先级更高,或者运算符栈为空,则将当前运算符压入运算符栈。
    • 否则,从运算符栈中弹出运算符,并从数字栈中弹出两个数字,执行运算,然后将结果压回数字栈。重复此过程,直到当前运算符可以被压入运算符栈。
  5. 处理括号:如果遇到左括号,直接压入运算符栈。如果遇到右括号,则执行运算,直到遇到左括号。

  6. 完成遍历:表达式遍历完成后,将运算符栈中剩余的运算符依次弹出,并执行运算。

个人思考与分析

在实现四则运算解析器的过程中,我深刻体会到了算法设计的重要性。算法不仅要正确,还要高效。在处理运算符优先级时,我选择了使用栈来管理运算符,这样可以避免复杂的条件判断,使得代码更加简洁。同时,我也意识到了算法的可扩展性。当前的实现只支持基本的四则运算,但通过增加新的运算符和相应的处理逻辑,可以轻松扩展到更复杂的表达式解析。

此外,我也思考了算法的健壮性。在实际应用中,用户输入的表达式可能包含错误,如不匹配的括号、非法字符等。因此,一个健壮的解析器应该能够处理这些异常情况,并给出合理的错误提示。这要求我们在设计算法时,不仅要关注正常情况,还要考虑异常情况的处理。

代码实现

以下是四则运算解析器的Python代码实现:

def calculate(expression):
    def precedence(op):
        if op in ('+', '-'):
            return 1
        if op in ('*', '/'):
            return 2
        return 0

    def apply_op(a, b, op):
        if op == '+': return a + b
        if op == '-': return a - b
        if op == '*': return a * b
        if op == '/': return int(a / b)  # 向下取整

    num_stack = []
    op_stack = []
    i = 0
    while i < len(expression):
        if expression[i].isdigit():
            num = 0
            while i < len(expression) and expression[i].isdigit():
                num = num * 10 + int(expression[i])
                i += 1
            num_stack.append(num)
        elif expression[i] in '+-*/':
            while (len(op_stack) != 0 and 
                   precedence(op_stack[-1]) >= precedence(expression[i])):
                op = op_stack.pop()
                num2 = num_stack.pop()
                num1 = num_stack.pop()
                num_stack.append(apply_op(num1, num2, op))
            op_stack.append(expression[i])
            i += 1
        elif expression[i] == ' ':
            i += 1
        elif expression[i] == '(':
            op_stack.append(expression[i])
            i += 1
        elif expression[i] == ')':
            while op_stack[-1] != '(':
                op = op_stack.pop()
                num2 = num_stack.pop()
                num1 = num_stack.pop()
                num_stack.append(apply_op(num1, num2, op))
            op_stack.pop()  # 弹出左括号
            i += 1

    while len(op_stack) != 0:
        op = op_stack.pop()
        num2 = num_stack.pop()
        num1 = num_stack.pop()
        num_stack.append(apply_op(num1, num2, op))

    return num_stack[0]

# 示例
expression = "3 + 2 * 2"
result = calculate(expression)
print(f"The result of '{expression}' is {result}")

复杂度分析

  • 时间复杂度O(n),其中n是表达式的长度。我们需要遍历整个表达式一次。
  • 空间复杂度O(n),用于存储数字栈和运算符栈。

总结

通过实现一个简单的四则运算解析器,我们不仅能够理解四则运算的解析和执行过程,还能够学习到栈这一数据结构的应用。这种解析器的实现方法可以推广到更复杂的表达式解析中,例如包含函数和变量的表达式。在实际应用中,四则运算解析器可以用于计算器程序、脚本语言解释器等多个领域。此外,实现过程中的思考也让我意识到了算法设计的复杂性和挑战,包括算法的正确性、效率、可扩展性和健壮性。从青训营中获取这些经验对于我未来的编程实践和算法学习都是非常宝贵的。