Leetcode 1352:最后K个数的乘积解法
参考 https://blog.csdn.net/qq_17550379/article/details/104348023 coordinate_blog大佬的解法
开始第一次直接使用的暴力,结果显示超时,代码就不说了。
第一种解法:学习了动态规划的方法,每一次添加,记录下前面的累积。最后计算的时候用
self.list0[-1]//self.list0[-k-1]
就能得到结果。
当然,如果中间出现0,就比较麻烦,然后就必须记录下0出现的位置。当然只记录下最后一个0出现的位置就可以了。
class ProductOfNumbers:
def __init__(self):
self.list0 = [1]
self.station=0
self.lenlist0=1 #记录一下长度
def add(self, num: int) -> None:
self.lenlist0+=1
if num==0:
self.list0.append(1)
self.station=self.lenlist0 #记录一下最后一个0出现的位置
else:
self.list0.append(num*self.list0[-1])
def getProduct(self, k: int) -> int:
# print(self.list0,self.lenlist0,self.station,k)
if k>self.lenlist0 or k>(self.lenlist0-self.station):
return 0
else:
return self.list0[-1]//self.list0[-k-1]
解法2:其实,只要是出现了0,那么list是可以重新启动的。代码更简洁了,运算量更小了。这是谁发现的规律,太牛了
class ProductOfNumbers:
def __init__(self):
self.list0 = [1]
def add(self, num: int) -> None:
if num:
self.list0.append(num*self.list0[-1])
else:
self.list0 = [1]
def getProduct(self, k: int) -> int:
if k>=len(self.list0):
return 0
else:
return self.list0[-1]//self.list0[-k-1]
浙公网安备 33010602011771号