统计特殊四元组

/*
 * 思路：
 * 使用 Java Streams API 不能直接处理嵌套循环，但我们可以利用 Streams 和组合索引来实现相同的逻辑。
 * 通过 IntStream.range 和组合函数进行索引组合，筛选符合条件的四元组。
 */

import java.util.stream.IntStream;

public class Solution {
    public long countQuadruplets(int[] nums) {
        int n = nums.length;
        return IntStream.range(0, n - 3)
                .boxed()
                .flatMap(a -> IntStream.range(a + 1, n - 2)
                        .boxed()
                        .flatMap(b -> IntStream.range(b + 1, n - 1)
                                .boxed()
                                .flatMap(c -> IntStream.range(c + 1, n)
                                        .filter(d -> nums[a] + nums[b] + nums[c] == nums[d])
                                        .boxed()))
                ).count();
    }
}