您的位置:首页 > 职场人生

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

2019-03-27 17:12 435 查看

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

字符串表达式仅包含非负整数,

+
, 
-
 ,
*
/
 四种运算符和空格 
 
。 整数除法仅保留整数部分。

示例 1:

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

示例 2:

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

示例 3:

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

说明:

  • 你可以假设所给定的表达式都是有效的。
  • 不要使用内置的库函数 
    eval
[code]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. 维护一个栈放数字,第一个运算符是+,在将每个数字入栈前,如果前面是+,则直接入栈,如果是-,则将该数字的相反数入栈。如果是*或/,则将栈顶数字和当前数字运算结果入栈。注意除法中要先处理负号的问题,还有要用//整除。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: