1
class ProductOfNumbers {
2
  List<Integer> prefix;
3

4
  public ProductOfNumbers() {
5
    prefix = new ArrayList<>();
6
    prefix.add(1);
7
  }
8

9
  public void add(int num) {
10
    if (num == 0) {
11
      prefix.clear();
12
      prefix.add(1);
13
    } else prefix.add(num * prefix.get(prefix.size() - 1));
14
  }
15

16
  public int getProduct(int k) {
17
    if (k >= prefix.size()) return 0;
18
    return prefix.get(prefix.size() - 1) / prefix.get(prefix.size() - k - 1);
19
  }
20
}