统计「优美子数组」

/*
 * 思路：
 * 我们需要找到包含恰好 k 个奇数的子数组的数量。
 * 通过使用Java Stream API，我们可以更优雅地实现这个问题。
 * 首先找到所有奇数的位置，然后计算有多少子数组可以包含这些奇数。
 * 对于每个窗口（包含 k 个奇数），我们可以通过计算其左边和右边的偶数数目来确定有多少个这样的子数组。
 */

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

public class BeautifulSubarraysStream {
    public int numberOfSubarrays(int[] nums, int k) {
        List<Integer> oddIndices = IntStream.range(0, nums.length)
                .filter(i -> nums[i] % 2 != 0)
                .collect(ArrayList::new, ArrayList::add, ArrayList::addAll);

        if (oddIndices.size() < k) {
            return 0;
        }

        return IntStream.range(0, oddIndices.size() - k + 1)
                .map(i -> {
                    int left = (i == 0) ? oddIndices.get(i) + 1 : oddIndices.get(i) - oddIndices.get(i - 1);
                    int right = (i + k == oddIndices.size()) ? nums.length - oddIndices.get(i + k - 1) : oddIndices.get(i + k) - oddIndices.get(i + k - 1);
                    return left * right;
                })
                .sum();
    }

    public static void main(String[] args) {
        BeautifulSubarraysStream bs = new BeautifulSubarraysStream();
        int[] nums = {1, 1, 2, 1, 1};
        int k = 3;
        System.out.println(bs.numberOfSubarrays(nums, k)); // 输出 2
    }
}