含最多 K 个可整除元素的子数组

/*
 * 思路：
 * 1. 使用 Java Stream 生成所有可能的子数组。
 * 2. 对每个子数组进行过滤，确保被 p 整除的元素个数不超过 k。
 * 3. 使用 Set 来存储这些子数组的字符串表示，确保唯一性。
 * 4. 返回 Set 的大小。
 */
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.IntStream;

public class Solution {
    public int countDistinctSubarrays(int[] nums, int k, int p) {
        Set<String> uniqueSubarrays = new HashSet<>();
        IntStream.range(0, nums.length).forEach(i ->
            IntStream.rangeClosed(i, nums.length - 1).forEach(j -> {
                int[] subarray = Arrays.copyOfRange(nums, i, j + 1);
                long count = Arrays.stream(subarray).filter(x -> x % p == 0).count();
                if (count <= k) uniqueSubarrays.add(Arrays.toString(subarray));
            })
        );
        return uniqueSubarrays.size();
    }
}