【Leetcode-算法】1352. 最后 K 个数的乘积（C++）

最新推荐文章于 2023-08-26 17:04:13 发布

原创

最新推荐文章于 2023-08-26 17:04:13 发布 · 366 阅读

收录于

这篇博客主要介绍了如何解决LeetCode中关于计算最后K个数乘积的问题，通过维护前缀积来优化算法，避免O(n)的时间复杂度。在遇到数字0时，需要进行特殊判断，若查询的k超过当前计数cnt，则答案为0。博主分享了AC代码并表示看到题解后的灵感。

请你实现一个「数字乘积类」ProductOfNumbers，要求支持下述两种方法：

1. add(int num)
将数字 num 添加到当前数字列表的最后面。

2. getProduct(int k)
返回当前数字列表中，最后 k 个数字的乘积。
你可以假设当前列表中始终 至少 包含 k 个数字。
题目数据保证：任何时候，任一连续数字序列的乘积都在 32-bit 整数范围内，不会溢出。

思路：

首先，存乘积而不存原数，此时是O(n)会超时，考虑维护前缀积来优化。用前缀积表示前i个数的乘积，答案即ans[cnt]/ans[cnt-k]。

遇到零进行额外的操作：如果查询的k大于cnt则直接返回0。

AC代码：

class ProductOfNumbers {
private:
    int ans[40005];
    int cnt;
public:
    ProductOfNumbers() {
        ans[0]=1;
        cnt=0;
    }
    
    void add(int num) {
        if(!num) cnt=0;
        else{
            ans[++cnt]=num;
            ans[cnt]*=ans[cnt-1];
        }
    }
    
    int getProduct(int k) {
        if(cnt<k) return 0;
        return ans[cnt]/ans[cnt-k];
    }
};

一些操作：

看题解才想到，觉得