统计好三元组

/*
 * 题目思路：
 * 给定一个整数数组 arr 和整数 a, b, c，要求统计其中满足以下条件的三元组的数量：
 * 1. 0 <= i < j < k < arr.length
 * 2. |arr[i] - arr[j]| <= a
 * 3. |arr[j] - arr[k]| <= b
 * 4. |arr[i] - arr[k]| <= c
 * 使用 Java Stream 处理。
 */
import java.util.stream.IntStream;

public class GoodTripletsStream {
    public long countGoodTriplets(int[] arr, int a, int b, int c) {
        return IntStream.range(0, arr.length - 2)
                .flatMap(i -> IntStream.range(i + 1, arr.length - 1)
                        .filter(j -> Math.abs(arr[i] - arr[j]) <= a)
                        .flatMap(j -> IntStream.range(j + 1, arr.length)
                                .filter(k -> Math.abs(arr[j] - arr[k]) <= b && Math.abs(arr[i] - arr[k]) <= c)))
                .count();
    }

    public static void main(String[] args) {
        GoodTripletsStream solution = new GoodTripletsStream();
        int[] arr = {3, 0, 1, 1, 9, 7};
        int a = 7, b = 2, c = 3;
        System.out.println(solution.countGoodTriplets(arr, a, b, c)); // 输出 4
    }
}