224. Basic Calculator

题目链接: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)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容