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]