224. Basic Calculator

https://leetcode.com/problems/basic-calculator/

题目解释

你会得到一个字串s ，这个字串是一组算式，请回传算式计算的结果

example

解法

这题比较麻烦的是它有括号需要处理，幸运的是他的字串不会出现乘除；因此，括号前面的负号是我们要烦恼的

这题的解法是用堆叠(stack)，把目前累积的结果和括号前面的正负号储存起来

等括号内的计算完成後，就把存起来的东西取出来并加在目前累积的结果

程序码

class Solution:
    def calculate(self, s: str) -> int:
        
        stack = []
        operation = '+' 
        num = 0
        ans = 0
        
        for i in range(len(s)):
            current = s[i]
            
            if current.isdigit():
                num = num * 10 + int(current)
            
            elif current in '+-':
                
                if operation == '+':
                    ans += num
                else:
                    ans -= num
                
                operation = current
                num = 0
            
            elif current == '(':
                stack.append(ans)
                stack.append(operation)
                operation = '+'
                num = 0
                ans = 0
            
            elif current == ')':
                
                if operation == '+':
                    ans += num
                else:
                    ans -= num
                
                operation = stack.pop()
                if operation == '-':
                    ans *= -1
                ans += stack.pop()
                num = 0
        
        
        if operation == '+':
            ans += num
        else:
            ans -= num
        return ans

不过这样的程序码不好修改成能处理乘除法

所以比较好的解答，应该是要在遇到括号时就呼叫自己(self.calculate)，并个别回传括号内的结果

闲聊

今天的题目因为不用处理乘除所以还算简单

大学刚学stack时也有练习过类似的题目

话说这两天有台风要来了，大家注意平安和防水罗!