leetcode-224-基本计算器

class Solution:
    def calculate(self, s: str) -> int:
        tokens = self.infix_to_suffix(s)
        result = self.evalRPN(tokens)
        return result
            
    def infix_to_suffix(self,s):
        s = re.split(‘(\W)‘,s)
        result = []
        stack = []
        op_priority = {‘*‘:2,
                      ‘/‘:2,
                      ‘%‘:2,
                      ‘+‘:1,
                      ‘-‘:1}
        for i in s:
            if i == ‘‘ or i == ‘ ‘:
                continue
            if i == ‘(‘:
                stack.append(i)
            elif i == ‘)‘:
                while stack[-1] != ‘(‘:
                    result.append(stack.pop())
                stack.pop()
            elif i in op_priority.keys():
                while stack and stack[-1] in op_priority.keys() and op_priority[i] <= op_priority[stack[-1]]:
                    result.append(stack.pop())
                stack.append(i)
            else:
                result.append(int(i))
        while stack:
            result.append(stack.pop())
        return result
    
    def evalRPN(self, tokens):
        f1 = lambda a,b:a+b
        f2 = lambda a,b: a - b
        f3 = lambda a,b: a*b
        f4 = lambda a,b: int(a/b)
        maps = {‘+‘:f1,‘-‘:f2,‘*‘:f3,‘/‘:f4}
        stack = []
        for token in tokens:
            if token in maps:
                a = stack.pop()
                b = stack.pop()
                stack.append(maps[token](b,a))
            else:
                token = int(token)
                stack.append(token)
        return stack[-1] 

class Solution:
    def calculate(self, s: str) -> int:
        res = 0
        stack = []
        sign = 1
        i = 0
        n = len(s)
        while i < n:
            if s[i] == " ":
                i += 1
            elif s[i] == "-":
                sign = -1
                i += 1
            elif s[i] == "+":
                sign = 1
                i += 1
            elif s[i] == "(":
                stack.append(res)
                stack.append(sign)
                res = 0
                sign = 1
                i += 1
            elif s[i] == ")":
                # print(stack)
                res = res * stack.pop() + stack.pop()
                i += 1
            elif s[i].isdigit():
                tmp = int(s[i])
                i += 1
                while i < n and s[i].isdigit():
                    tmp = tmp * 10 + int(s[i])
                    i += 1
                res += tmp * sign
        return res