美观的花束

/*
 * 思路：使用Java Stream来实现这个算法需要一些变通的方法，因为Stream并不直接支持滑动窗口操作。
 * 我们可以先将数据进行处理，然后再使用Stream来进行结果的计算。
 */
import java.util.HashMap;
import java.util.stream.IntStream;

public class BeautifulBouquetStream {
    public int countBeautifulBouquets(int[] flowers, int cnt) {
        final int MOD = 1000000007;
        int n = flowers.length;
        HashMap<Integer, Integer> countMap = new HashMap<>();
        int[] resultArray = new int[n];

        int left = 0;
        for (int right = 0; right < n; right++) {
            countMap.put(flowers[right], countMap.getOrDefault(flowers[right], 0) + 1);
            while (countMap.get(flowers[right]) > cnt) {
                countMap.put(flowers[left], countMap.get(flowers[left]) - 1);
                left++;
            }
            resultArray[right] = right - left + 1;
        }

        return IntStream.of(resultArray).reduce(0, (sum, value) -> (sum + value) % MOD);
    }

    public static void main(String[] args) {
        BeautifulBouquetStream solution = new BeautifulBouquetStream();
        int[] flowers1 = {1, 2, 3, 2};
        int cnt1 = 1;
        System.out.println(solution.countBeautifulBouquets(flowers1, cnt1)); // 输出：8

        int[] flowers2 = {5, 3, 3, 3};
        int cnt2 = 2;
        System.out.println(solution.countBeautifulBouquets(flowers2, cnt2)); // 输出：8
    }
}