题干
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
方法一:动态规划
本问题可以用动态规划求解。
给定一个有序序列 1 … n,为了根据序列构建一棵二叉搜索树。我们可以遍历每个数字 i,将该数字作为树根,1 … (i-1) 序列将成为左子树,(i+1) … n 序列将成为右子树。于是,我们可以递归地从子序列构建子树。
在上述方法中,由于根各自不同,每棵二叉树都保证是独特的。
可见,问题可以分解成规模较小的子问题。因此,我们可以存储并复用子问题的解,而不是递归的(也重复的)解决这些子问题,这就是动态规划法。
算法
问题是计算不同二叉搜索树的个数。为此,我们可以定义两个函数:
-
G ( n ) G(n) G(n): 长度为n的序列的不同二叉搜索树个数。
-
F ( i , n ) F(i,n) F(i,n): 以i为根的不同二叉搜索树个数( 1 ≤ i ≤ n 1≤i≤n 1≤i≤n)。
可见,G(n) 是我们解决问题需要的函数。
稍后我们将看到, G ( n ) G(n) G(n) 可以从 F ( i , n ) F(i,n) F(i,n)得到,而 F ( i , n ) F(i,n) F(i,n)又会递归的依赖于 G ( n ) G(n) G(n)。
首先,根据上一节中的思路,不同的二叉搜索树的总数 G ( n ) G(n) G(n),是对遍历所有 i
(1 <= i <= n) 的