所有可能的真二叉树

/*
 * 思路：
 * - 同样是递归生成真二叉树，只是使用Java Stream的方式来实现。
 * - 通过flatMap将递归结果展平，并使用collect进行结果的收集。
 */

import java.util.*;
import java.util.stream.*;

class TreeNode {
    int val;
    TreeNode left, right;
    TreeNode(int val) { this.val = val; }
}

public class Solution {
    public List<TreeNode> allPossibleFBT(int n) {
        if (n % 2 == 0) return new ArrayList<>();
        if (n == 1) return Collections.singletonList(new TreeNode(0));
        n--;
        return IntStream.range(1, n)
                        .filter(i -> i % 2 == 1)
                        .mapToObj(i -> allPossibleFBT(i)
                                      .stream()
                                      .flatMap(left -> allPossibleFBT(n - i)
                                                     .stream()
                                                     .map(right -> {
                                                         TreeNode root = new TreeNode(0);
                                                         root.left = left;
                                                         root.right = right;
                                                         return root;
                                                     })))
                        .flatMap(x -> x)
                        .collect(Collectors.toList());
    }
}