Leetcode 1352：最后K个数的乘积解法

Posted 2021-03-08 耀扬

参考 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]