python leetcode 227. Basic Calculator II

最新推荐文章于 2021-03-11 17:58:04 发布

Neekity

最新推荐文章于 2021-03-11 17:58:04 发布

阅读量192

点赞数

分类专栏： leetcode python 文章标签： python leetcode

本文链接：https://blog.csdn.net/Neekity/article/details/85135200

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

class Solution(object):
    def calculate(self, s):
        """
        :type s: str
        :rtype: int
        """
        def helper(op, val):
            if op == '+':
                stack.append(val)
            elif op == '-':
                stack.append(-val)
            elif op == '*':
                stack.append(stack.pop() * val)
            elif op == '/':
                l, r = stack.pop() , val
                if l * r < 0 and l % r != 0:
                    stack.append(l // r  + 1)
                else:
                    stack.append(l // r)
        
        stack, val, op = [], 0, '+'
        
        for i in range(len(s)):
            if s[i].isdigit():
                val = val * 10 + int(s[i])
            elif s[i] in '/*-+':
                helper(op, val)
                op, val = s[i], 0
                
        helper(op, val)
                
        return sum(stack)