将数组划分成若干好子数组的方式

/*
 * 给定一个二元数组 nums。
 * 如果某个子数组恰好只存在一个值为 1 的元素，则认为该子数组是一个好子数组。
 * 要统计将数组 nums 划分成若干好子数组的方法数，并返回其对 10^9 + 7 取余之后的结果。
 * 
 * 思路：
 * 1. 使用 Java Stream 记录所有 1 的位置。
 * 2. 计算每两个相邻 1 之间的距离，然后可以选择插入这些位置的好子数组数目。
 * 3. 最后将所有的可能性相乘，即为结果。
 */

import java.util.*;
import java.util.stream.*;

public class GoodSubarraysStream {
    public int numGoodSubarrays(int[] nums) {
        final int MOD = 1000000007;
        List<Integer> onesPositions = IntStream.range(0, nums.length)
                                               .filter(i -> nums[i] == 1)
                                               .boxed()
                                               .collect(Collectors.toList());
        if (onesPositions.size() == 0) return 0;
        long result = IntStream.range(1, onesPositions.size())
                               .mapToLong(i -> onesPositions.get(i) - onesPositions.get(i - 1))
                               .reduce(1, (a, b) -> (a * b) % MOD);
        return (int) result;
    }

    public static void main(String[] args) {
        GoodSubarraysStream gs = new GoodSubarraysStream();
        int[] nums1 = {0, 1, 0, 0, 1};
        System.out.println(gs.numGoodSubarrays(nums1)); // 输出: 3

        int[] nums2 = {0, 1, 0};
        System.out.println(gs.numGoodSubarrays(nums2)); // 输出: 1
    }
}