【Leetcode】计算器

2023-12-18 19:25:57

思路

用栈来完成；

  • 考虑到运算关系，先乘除后加减；
  • 此外，一般计算式首个数字式正数；
  • 判断字符是否为数字，str.isdigit()
  • 字符转数字：ord(str) - ord(‘0’)
  • 遇到加减符，压栈数字；遇到乘除符，先运算再压栈
  • 遍历完后，再将栈内数字求和

代码

class Solution:
    def calculate(self, s: str) -> int:
        stack = []
        n = len(s)
        preSign = '+'
        num = 0
        for i in range(n):
            if s[i] != ' ' and s[i].isdigit():
                num = num * 10 + ord(s[i]) - ord('0')

            if i == n - 1 or s[i] in '+-*/':
                if preSign == '+':
                    stack.append(num)
                elif preSign == '-':
                    stack.append(-num)
                elif preSign == '*':
                    stack.append(stack.pop() * num)
                else:
                    stack.append(stack.pop // num)
                
                preSign = s[i]
                num = 0
        
        return sum(stack)

文章来源:https://blog.csdn.net/weixin_41845840/article/details/135065456
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。