95. 不同的二叉搜索树 II（Unique Binary Search Trees II）

题解

卡特兰数问题

递归

二叉排序树的重要性质是，左子树上的节点值一定小于根节点，右子树上的节点值一定大于根节点。由此，可知若根节点为$i$，则左子树的节点为$[1,...,i-1]$，右子树的节点为$[i+1,...,n]$。可递归的构建二叉搜索树。

1. 特判，若$n==0$，返回$[]$

2. 定义生成树函数$build\_Trees(left,right)$，表示生成$[left,...,right]$的所有可能二叉搜索树：

   • 若$left>right$，说明为空树，返回$[None]$
   • 初始化所有可能的二叉搜索树列表$all\_trees=[]$
   • 遍历每一种可能的节点$i$，遍历区间$[left,right+1)$:
     • 所有可能的左子树列表$left\_trees=build\_Trees(left,i-1)$
     • 所有可能的右子树列表$right\_trees=build\_Trees(i+1,right)$
     • 组合所有的方式，遍历左子树列表,$l$：遍历右子树列表,$r$
       • 建立当前树节点$cur\_tree=TreeNode(i)$
       • 将$cur\_tree$左子树置为$l$
       • 将$cur\_tree$右子树置为$r$
       • 将$cur\_tree$加入树列表中
   • 返回树列表$all\_trees$

3. 返回$build\_Trees(1,n)$

复杂度分析

• 时间复杂度：$O(\frac{4^n}{n^{1/2}})$
• 空间复杂度：$O(\frac{4^n}{n^{1/2}})$

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def generateTrees(self, n: int) -> List[TreeNode]:
        if(n==0):
            return []
        def build_Trees(left,right):
            all_trees=[]
            if(left>right):
                return [None]
            for i in range(left,right+1):
                left_trees=build_Trees(left,i-1)
                right_trees=build_Trees(i+1,right)
                for l in left_trees:
                    for r in right_trees:
                        cur_tree=TreeNode(i)
                        cur_tree.left=l
                        cur_tree.right=r
                        all_trees.append(cur_tree)
            return all_trees
        res=build_Trees(1,n)
        return res

Java(待完成)