码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-02-22 23:41:54      阅读:85      评论:0      收藏:0      [点我收藏+]

标签:pen   station   int   bsp   规律   ati   highlight   开始   else   

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

  

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

标签:pen   station   int   bsp   规律   ati   highlight   开始   else   

原文地址:https://www.cnblogs.com/duoba/p/12347964.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!