今天滴滴第一面遇到这道题,mark一下!
初级版
题目
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式仅包含非负整数,+, - ,*,/ 四种运算符和空格 。 整数除法仅保留整数部分。
示例
输入: "3+2*2"
输出: 7
输入: " 3/2 "
输出: 1
输入: " 3+5 / 2 "
输出: 5
解题思路
用栈的思想实现,当op为‘+’、‘-’时,把数字放入栈,当op为‘*’、‘/’时,把栈里的最后一个数弹出,然后做运算之后再放进去,最后求该栈的和。
- 判断是否输入为空,为空则返回0;
- 初始化一个op为运算符‘+’或‘-’,‘*’、‘/’不行;
- 用一个循环求第一个数字num;
- op为‘+’, num入栈。
- op为‘-’, -num入栈。
- op为‘’, numres.pop()入栈。
- op为‘/’,这里要注意,题目要整除,当负数时,整除的进位是错的,例如-3/2=2;因此要先判断tmp是否为负,如果为负则先把tmp转为整数,再把整个结果转为负数。
- 注意有可能字符串中有空格,要额外判断下。
- 最后直接求和,因为此时只剩下‘+’、‘-’,而且数字已经处理好正负,直接求和即可。
class Solution:
def calculate(self, s: str) -> int:
if len(s)==0:
return 0
res=[]
i=0
op='+'
while(i<len(s)):
if s[i].isdigit():
num=0
while i<len(s) and s[i].isdigit():
num=num*10+int(s[i])
i+=1
if op=='+':
res.append(num)
if op=='-':
res.append(-num)
if op=='*':
tmp=res.pop()
res.append(tmp*num)
if op=='/':
tmp=res.pop()
if tmp<0:
res.append(-(-tmp//num))
else:
res.append(tmp//num)
elif s[i] != ' ' :
op=s[i]
i+=1
else:
i+=1
return sum(res)
升级版
题目
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式可以包含左括号 ( ,右括号 ),加号 + ,减号 -,非负整数和空格 。
解题思路
刚实现了一个初级版的,然后遇到带括号的,就会很自然想到截取出括号里面的子字符串直接调用刚刚实现的函数,但是超时了。
class Solution:
def calculatetmp(self, s: str) -> int:
if len(s)==0:
return 0
res=[]
i=0
op='+'
while(i<len(s)):
if s[i].isdigit():
num=0
while i<len(s) and s[i].isdigit():
num=num*10+int(s[i])
i+=1
if op=='+':
res.append(num)
if op=='-':
res.append(-num)
if op=='*':
tmp=res.pop()
res.append(tmp*num)
if op=='/':
tmp=res.pop()
if tmp<0:
res.append(-(-tmp//num))
else:
res.append(tmp//num)
elif s[i] != ' ' :
op=s[i]
i+=1
else:
i+=1
return sum(res)
def calculate(self, s: str) -> int:
if len(s)==0:
return 0
i=0
j=-1
while len(s)>i:
# print(s[i])
if s[i]==')':
j=i
while j >= 0:
if s[j]=='(':
tmpres=s[:j]+str(self.calculatetmp(s[j+1:i]))+s[i+1:]
s=tmpres
break
j-=1
if j!=-1:
i=j
else:
i+=1
return self.calculatetmp(s)
去看了大神的解法,如此优美易懂!!!呜呜呜我好菜啊
还是觉得第一种做法也是可行的,然后今天优化了一下,当遇到左括号时记录当时的坐标,当遇到右括号时,弹出当时的坐标即可取出括号里的字符。
运行通过了,但是时间和内存开销比较大。
# class Solution:
# def calculate(self, s):
# res, num, sign, stack = 0, 0, 1, []
# for ss in s:
# if ss.isdigit():
# num = 10*num + int(ss)
# elif ss in ["-", "+"]:
# res += sign*num
# num = 0
# sign = [-1, 1][ss=="+"]
# elif ss == "(":
# stack.append(res)
# stack.append(sign)
# sign, res = 1, 0
# elif ss == ")":
# res += sign*num
# res *= stack.pop()
# res += stack.pop()
# num = 0
# return res + num*sign
class Solution:
def calculatetmp(self, s: str) -> int:
if len(s)==0:
return 0
res=[]
i=0
op='+'
while(i<len(s)):
if s[i].isdigit():
num=0
while i<len(s) and s[i].isdigit():
num=num*10+int(s[i])
i+=1
if op=='+':
res.append(num)
if op=='-':
res.append(-num)
if op=='*':
tmp=res.pop()
res.append(tmp*num)
if op=='/':
tmp=res.pop()
if tmp<0:
res.append(-(-tmp//num))
else:
res.append(tmp//num)
elif s[i] != ' ' :
op=s[i]
i+=1
else:
i+=1
return sum(res)
def calculate(self, s: str) -> int:
if len(s)==0:
return 0
i=0
j=-1
indx = []
while len(s)>i:
# print(s,indx,s[i])
if s[i]=='(':
indx.append(i)
if s[i]==')':
j = indx.pop()
if self.calculatetmp(s[j+1:i])<0 and s[j-1]=='-':
tmpres=s[:j-1]+'+'+str(abs(self.calculatetmp(s[j+1:i])))+s[i+1:]
else:
tmpres=s[:j]+str(self.calculatetmp(s[j+1:i]))+s[i+1:]
s=tmpres
i=j-1
i+=1
# print(s)
return self.calculatetmp(s)
大神的解法虽然优美,但是仅仅实现了加减括号,没有乘除。
class Solution:
def calculate(self, s):
res, num, sign, stack = 0, 0, 1, []
for ss in s:
if ss.isdigit():
num = 10*num + int(ss)
elif ss in ["-", "+"]:
res += sign*num
num = 0
sign = [-1, 1][ss=="+"]
elif ss == "(":
stack.append(res)
stack.append(sign)
sign, res = 1, 0
elif ss == ")":
res += sign*num
res *= stack.pop()
res += stack.pop()
num = 0
return res + num*sign
leetcode还有个更高级的,应该是要实现带加减乘除和括号的计算器。不过我不是会员,看不到具体题目,也想不出来。
第二天,想粗来啦!!!!再想想怎么优化!!!!