实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式可以包含左括号 ( ,右括号 ),加号 + ,减号 -,非负整数和空格 。
示例 1:
输入: "1 + 1"
输出: 2
示例 2:
输入: " 2-1 + 2 "
输出: 3
示例 3:
输入: "(1+(4+5+2)-3)+(6+8)"
输出: 23
说明:
你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/basic-calculator
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
一般问括号都是栈,此题同理。
难点在于处理括号外的符号,比如 3+5-(3+2)中的负号,会对整个括号内的值产生影响。
所以可以用一个栈记录括号外的符号,和括号之前的运算结果。
class Solution(object):
def calculate(self, s):
"""
:type s: str
:rtype: int
"""
res = 0
num = 0
stack = []
sign = 1
for c in s:
if c.isdigit():
num = num * 10 + int(c)
elif c == "+":
res += sign * num
num = 0
sign = 1
elif c == "-":
res += sign * num
num = 0
sign = -1
elif c =="(":
stack.append(res)
stack.append(sign)
sign = 1
res = 0
elif c == ")":
res += num * sign
num = 0
res = res * stack.pop() + stack.pop()
res += num * sign
return res