題目鏈接:224
思路
想象將所有的括號全部去掉,然后進行計算。去掉括號時要根據(jù)括號外的符號看是否需要改變括號內(nèi)的符號。
func calculate(s string) int {
ops := []int{1} // 一個棧。記錄括號前的符號,用于決定括號中每個操作符是否需要變
sign := 1 // 記錄下一個“數(shù)字”前面的操作符
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
}
時間復(fù)雜度:O(n)
空間復(fù)雜度:O(n)