题目分析:
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?示例:输入: 3,输出: 5
解释:给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
解题思路:
对于一点i,当其为根节点时,左子树的节点的个数为i-1,(为1,…i-1),右子树的个数为n-i(为,i+1,…n)。对一个根来说,唯一二叉树的个数为左子树结点的个数乘以右子树的个数。而根节点可以从1到n 中选择。我们可以容易用递归实现,但是提交时间超限。可以进一步思考使用动态规划,基本思路是一致的。
如n = 3时:
dp[3] = dp[0] * dp[2]
dp[3] = dp[1] * dp[1]
dp[3] = dp[2] * dp[0]
提交代码:(动态规划,Runtime: 36 ms, faster than 74.08% )
class Solution:
def numTrees(self, n: int) -> int:
dp = [0 for i in range(n + 1)]
dp[0] = dp[1] = 1
for i in range(2, n + 1):
for j in range(i):
dp[i] += dp[j] * dp[i - j - 1]
return dp[n]
测试代码1:(递归,注意时间超限TLE,不能提交)
class Solution:
def numTrees(self, n: int) -> int:
if n <= 1:
return 1
sum = 0
for i in range(1, n + 1):
sum += self.numTrees(i - 1) * self.numTrees(n - i)
return sum
print(Solution().numTrees(3))