统计 K-Free 子集的总数

/*
 * 思路：
 * 1. 创建一个函数用于判断一个子集是否为K-Free。
 * 2. 使用流生成所有可能的子集并过滤掉不满足条件的子集。
 * 3. 对过滤后的子集计数。
 */

import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Solution {
    public long countKFreeSubsets(int[] nums, int k) {
        List<List<Integer>> subsets = IntStream.range(0, 1 << nums.length)
            .mapToObj(i -> IntStream.range(0, nums.length)
                .filter(j -> (i & (1 << j)) != 0)
                .mapToObj(j -> nums[j])
                .collect(Collectors.toList()))
            .collect(Collectors.toList());
        return subsets.stream()
            .filter(subset -> isKFree(subset, k))
            .count();
    }

    private boolean isKFree(List<Integer> subset, int k) {
        for (int i = 0; i < subset.size(); i++) {
            for (int j = i + 1; j < subset.size(); j++) {
                if (Math.abs(subset.get(i) - subset.get(j)) == k) {
                    return false;
                }
            }
        }
        return true;
    }
}