1、题目描述
https://leetcode-cn.com/problems/unique-binary-search-trees/
给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
相关题:95. 不同的二叉搜索树 II(打印出树) https://blog.csdn.net/IOT_victor/article/details/107067866
2、代码详解
动态规划
卡特兰数 公式 https://baike.baidu.com/item/卡特兰数
G(n)=G(0)∗G(n−1)+G(1)∗G(n−2)+...+G(n−1)∗G(0)
class Solution(object):
def numTrees(self, n):
"""
:type n: int
:rtype: int
"""
dp = [0] * (n + 1)
dp[0] = 1
dp[1] = 1
for i in range(2, n + 1):
for j in range(1, i + 1):
dp[i] += dp[j - 1] * dp[i - j]
return dp[n]
n = 3
s = Solution()
print(s.numTrees(n)) # 5
给定一个有序序列 1 ... n,为了根据序列构建一棵二叉搜索树。
我们可以遍历每个数字 i,将该数字作为树根,1 ... (i-1) 序列将成为左子树,(i+1) ... n 序列将成为右子树。于是,我们可以递归地从子序列构建子树。在上述方法中,由于根各自不同,每棵二叉树都保证是独特的。
定义两个函数:
- G(n): 长度为n的序列的不同二叉搜索树个数。 是我们解决问题需要的函数
- F(i, n): 以i为根的不同二叉搜索树个数(1≤i≤n)。
边界情况,当序列长度为 1 (只有根)或为 0 (空树)时,G(0)=1,G(1)=1