LeetCode-Solutions/Solutions/1352.java at main · MohamedMetwalli5/LeetCode-Solutions · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
class ProductOfNumbers {
    ArrayList<Integer> prefixProduct;

    public ProductOfNumbers() {
        prefixProduct = new ArrayList();
    }

    public void add(int num) {
        if(num == 0){
            prefixProduct.clear();
        }else{
            if(prefixProduct.size() == 0){
                prefixProduct.add(num);
            }else{
                prefixProduct.add(num * prefixProduct.get(prefixProduct.size()-1));
            }
        }
    }

    public int getProduct(int k) {
        if(k > prefixProduct.size()){
            return 0;
        }
        if(prefixProduct.size() == k){
            return prefixProduct.get(prefixProduct.size()-1);
        }
        return (prefixProduct.get(prefixProduct.size()-1) / prefixProduct.get(prefixProduct.size()-k-1));
    }
}

/**
 * Your ProductOfNumbers object will be instantiated and called as such:
 * ProductOfNumbers obj = new ProductOfNumbers();
 * obj.add(num);
 * int param_2 = obj.getProduct(k);
 */

/**
 * Your ProductOfNumbers object will be instantiated and called as such:
 * ProductOfNumbers obj = new ProductOfNumbers();
 * obj.add(num);
 * int param_2 = obj.getProduct(k);
 */