组合

/*
 * 思路：虽然Java Stream不是最适合回溯算法的工具，但我们可以使用组合的方式来生成所有可能的组合。
 * 这里我们使用一个辅助函数来生成组合并转换为流。
 */
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SolutionStream {
    public List<List<Integer>> combine(int n, int k) {
        return combineHelper(IntStream.rangeClosed(1, n).boxed().collect(Collectors.toList()), k);
    }

    private List<List<Integer>> combineHelper(List<Integer> nums, int k) {
        if (k == 0) {
            List<List<Integer>> result = new ArrayList<>();
            result.add(new ArrayList<>());
            return result;
        }
        if (nums.isEmpty()) return new ArrayList<>();

        int head = nums.get(0);
        List<Integer> rest = nums.subList(1, nums.size());

        List<List<Integer>> excludingHead = combineHelper(rest, k);
        List<List<Integer>> includingHead = combineHelper(rest, k - 1).stream()
                .peek(list -> list.add(0, head))
                .collect(Collectors.toList());

        List<List<Integer>> result = new ArrayList<>(excludingHead);
        result.addAll(includingHead);
        return result;
    }
}