leetcode [227] 基本计算器 II
Contact me:
Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式仅包含非负整数,+, - ,*,/ 四种运算符和空格 。 整数除法仅保留整数部分。
示例 1:
输入: "3+2*2"
输出: 7
示例 2:
输入: " 3/2 "
输出: 1
示例 3:
输入: " 3+5 / 2 "
输出: 5
说明:
你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。
来自题解:
使用一个栈存储数字,遍历字符,遇到数字,历史值*10再相加,如果遇到运算符号,加减直接添加,乘除计算栈顶。注意,op记录上次的运算符号,遇到符号把历史值清0。
class Solution:
def calculate(self, s: str) -> int:
num = 0
stack = list()
op = '+'
for i, c in enumerate(s):
if c.isnumeric():
num = num*10+int(c)
if c in '+-*/' or i == len(s) - 1:
if op == '+':
stack.append(num)
if op == '-':
stack.append(-num)
if op == '*':
stack.append(stack.pop()*num)
if op == '/':
stack.append(int(stack.pop()/num))
op = c
num = 0
return sum(stack)