【Leetcode】1352. 最后 K 个数的乘积

博文作者 wangzirui32
喜欢的可以 点赞 收藏 关注哦~~
本文首发于CSDN，未经许可禁止转载

1. 题目描述

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/product-of-the-last-k-numbers/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题思路

这道题有点难度，我们不能记录每次add添加的数字，这样每次getProduct都要计算乘积，耗时较大（我用此方法编写代码显示超时），于是，我们需要换一种思路：我们可不可以每次add时顺带把乘积计算完毕存入列表呢？可以，假设我们输入了如下长度为4的列表：

索引	0	1	2	3
数值	$a$	$b$	$c$	$d$

然后，我们把它转化为一个新列表：

索引	0	1	2	3
数值	$a$	$ab$	$abc$	$abcd$

可以看到，数值依次成为了ab， abc等的乘积，设我们需要计算后K个数的乘积（假设$k=2$），等价于求$cd$的乘积，就是 $abcd/ab=cd$，也就是索引倒数第一除以倒数第三，引入$K$即可表达为：
$$result=nums[-1]/nums[-k-1]$$

3. 代码实现

Code：

class ProductOfNumbers:
    def __init__(self):
        self.product_list = [1]  # 初始值为1

    def add(self, num):
        if num == 0:     # 如果为0 在此之后的乘积均为0 所以重置列表
        	self.product_list = [1]
        else:
        	# 与列表最后一项相乘 即可出现 a ab abc 一类的列表
        	self.product_list.append(num * self.product_list[-1])

    def getProduct(self, k):
        if k >= len(self.product_list):
        	# k大于长度返回0
        	return 0
        else:
        	# result = nums[-1] / nums[-k-1] 
        	return self.product_list[-1] / self.product_list[-k-1]

4. 最终结果

好了，今天的课程就到这里，我是wangzirui32，喜欢的可以点个收藏和关注，我们下次再见！