leetcode 刷题记录（高频算法面试题汇总）--基本计算器 II

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格  。 整数除法仅保留整数部分。

示例 1:

输入: "3+2*2"
输出: 7

示例 2:

输入: " 3/2 "
输出: 1

示例 3:

输入: " 3+5 / 2 "
输出: 5

说明：

• 你可以假设所给定的表达式都是有效的。
• 请不要使用内置的库函数 eval。

class Solution:
    def calculate(self, s: str) -> int:
        stack = []
        op = '+'
        i = 0
        
        while i<len(s):
            if s[i].isdigit():
                num = 0
                while i<len(s) and s[i].isdigit():
                    num = num*10 + ord(s[i]) - ord('0')
                    i += 1
                if op == '+':
                    stack.append(num)
                if op == '-':
                    stack.append(-num)
                if op == '*':
                    stack.append(num*stack.pop())
                if op == '/':
                    tmp = stack.pop()
                    if tmp < 0:
                        stack.append(-(-tmp//num))
                    else:
                        stack.append(tmp//num)
                continue
            elif s[i] != ' ':
                op = s[i]
            i += 1
        return sum(stack)

问题&思路：

1. 维护一个栈放数字，第一个运算符是+，在将每个数字入栈前，如果前面是+，则直接入栈，如果是-，则将该数字的相反数入栈。如果是*或/，则将栈顶数字和当前数字运算结果入栈。注意除法中要先处理负号的问题，还有要用//整除。