不同的二叉搜索树 IIPython解法

给你一个整数 n ，请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii

例：

输入：n = 3
输出：[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]]

解析：

递归，分成左子树和右子树两个子问题。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def generateTrees(self, n):
        """
        :type n: int
        :rtype: List[TreeNode]
        """
        res = []  # 存放最终结果
        if n < 1:
            return res
        else:
            return self.generateBST(1, n)
        
    def generateBST(self, start, end):
        res = []  # 存放临时子树
        if start > end:
            res.append(None)
            return res
        else:
            for i in range(start, end + 1):
                # 左子树
                left_tree = self.generateBST(start, i - 1)  # 添加所有不同的左子树
                # 右子树
                right_tree = self.generateBST(i + 1, end)  # 添加所有不同的右子树
                for left in left_tree:
                    for right in right_tree:  # 左右子树组合回到上一层的递归
                        root = TreeNode(i)
                        root.left = left
                        root.right = right
                        res.append(root)
            return res  # 返回上一层递归的左/右子树结果