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

参考 https://blog.csdn.net/qq_17550379/article/details/104348023 大佬的解法

开始第一次直接使用的暴力,结果显示超时,代码就不说了。

第一种解法:学习了动态规划的方法,每一次添加,记录下前面的累积。最后计算的时候用

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]

  

猜你喜欢

转载自www.cnblogs.com/duoba/p/12347964.html