可被整除的三元组数量

/*
 * 思路:
 * 1. 使用Java Stream遍历数组中的所有可能的三元组。
 * 2. 过滤出和可以被目标值整除的三元组。
 * 3. 计数所有符合条件的三元组。
 */

import java.util.stream.IntStream;

public class DivisibleTripletsStream {
    public long countDivisibleTriplets(int[] nums, int target) {
        return IntStream.range(0, nums.length - 2)
                .boxed()
                .flatMap(i -> IntStream.range(i + 1, nums.length - 1)
                        .boxed()
                        .flatMap(j -> IntStream.range(j + 1, nums.length)
                                .filter(k -> (nums[i] + nums[j] + nums[k]) % target == 0)
                                .mapToObj(k -> new int[]{i, j, k})))
                .count();
    }
}