给定一个整数 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
思路
二叉搜索树的子树也是 二叉搜索树
首先根据n建立数组s,储存每个节点建立时的val
数组s能生成的所有二叉搜索树,为其中每个元素分别作为根节点所生成的所有树的总和
对于数组s中的任意元素s[i],以s[i]的值为根节点生成的二叉搜索树,其左子树的所有结构是由s[:i]生成,右子树的所有结构是由s[i+1:]生成,将所有情况结合起来
递归即可
代码
def generateTrees(self, n):
if(n==0):return []
s=[_ for _ in range(n) ]
return self.f(s)
def f(self,s):
if(len(s)==1):return [TreeNode(s[0]+1)]
if(s==[]):return [None]
l=[]
for i in range(len(s)):
for a in self.f(s[:i]):
for b in self.f(s[i+1:]):
N=TreeNode(s[i]+1)
N.left=a
N.right=b
l.append(N)