leetcode-95. 不同的二叉搜索树 II

题目

给定一个整数 n，生成所有由 1 … n 为节点所组成的 二叉搜索树 。

示例：

输入：3
输出：
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
解释：
以上的输出对应以下 5 种不同结构的二叉搜索树：

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

提示：

0 <= n <= 8

解题思路

先看简单版：不同的二叉搜索树，不需要求出具体的树，只需要求数字，则可以简化。本题需要求出具体的树

递归，对1,2,3节点们，先以1作为根节点，然后递归得到后面的数字生成的子树，再以2为根节点，先递归得到左边的数字的作为左子树，再递归得到右边数字作为右子树，再对左右子树的根节点做笛卡尔积得到所有可能的pair

时间复杂度应该有$o(n^3)$

代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self, ):
        self.memo = {}

    def helper(self, available_numbers: tuple) -> List[Optional[TreeNode]]:
        if len(available_numbers) == 1:
            return [TreeNode(available_numbers[0])]
        if len(available_numbers) == 0:
            return [None]
        if available_numbers in self.memo:
            return self.memo[available_numbers]
        res = []
        for i, item in enumerate(available_numbers):
            left_children = self.helper(available_numbers[:i])
            right_children = self.helper(available_numbers[i + 1:])
            for each_l in left_children:
                for each_r in right_children:
                    root = TreeNode(item)
                    root.left = each_l
                    root.right = each_r
                    res.append(root)
        self.memo[available_numbers] = res
        return res

    def generateTrees(self, n: int) -> List[Optional[TreeNode]]:
        return self.helper(tuple(range(1, n + 1)))