水果成篮

/*
 * 题目思路：
 * 使用Java Stream API处理滑动窗口问题。
 * 通过收集所有可能的子数组长度，其中每个子数组最多包含两种不同的水果，
 * 然后找出最大的子数组长度。
 */
import java.util.*;
import java.util.stream.*;

public class Solution {
    public int totalFruit(int[] fruits) {
        return IntStream.range(0, fruits.length)
                .map(start -> {
                    Map<Integer, Long> count = new HashMap<>();
                    return IntStream.range(start, fruits.length)
                            .takeWhile(end -> {
                                count.put(fruits[end], count.getOrDefault(fruits[end], 0L) + 1);
                                return count.size() <= 2;
                            })
                            .count();
                })
                .max()
                .orElse(0);
    }
}