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

请你实现一个「数字乘积类」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];
    }
};

一些操作