最后 K 个数的乘积

/* 思路: 使用Java Stream API进行流式处理。我们将通过将所有数字存储在列表中并按需计算乘积。Java的Stream API提供了一种方便的方式来处理集合数据。 */

import java.util.ArrayList;
import java.util.List;
import java.util.stream.IntStream;

public class ProductOfNumbers {
    private List<Integer> nums;

    public ProductOfNumbers() {
        nums = new ArrayList<>();
    }

    public void add(int num) {
        if (num == 0) {
            nums.clear();
        }
        nums.add(num);
    }

    public int getProduct(int k) {
        if (nums.size() < k) return 0;
        return IntStream.range(nums.size() - k, nums.size())
                        .map(nums::get)
                        .reduce(1, (a, b) -> a * b);
    }
}