题目链接:224
思路
想象将所有的括号全部去掉,然后进行计算。去掉括号时要根据括号外的符号看是否需要改变括号内的符号。
func calculate(s string) int {
ops := []int{1} // 一个栈。记录括号前的符号,用于决定括号中每个操作符是否需要变
sign := 1 // 记录下一个“数字”前面的操作符
i := 0
ans := 0
for i < len(s) {
switch s[i] {
case ' ':
i++
case '+':
sign = ops[len(ops) - 1]
i++
case '-':
sign = -ops[len(ops) - 1]
i++
case '(':
ops = append(ops, sign)
i++
case ')':
ops = ops[: len(ops) - 1]
i++
default:
num := 0
for i < len(s) && s[i] >= '0' && s[i] <= '9' {
num = 10 * num + int(s[i] - '0')
i++
}
ans += sign * num
}
}
return ans
}
时间复杂度:O(n)
空间复杂度:O(n)