题目描述
原题地址
思路
由于每一个序列节点都有可能成为根节点,所以在遍历每个序列节点时,显然会出现重复计算左右子树的种类个数的情况,因此可以想到用DP来解决,避免大量的重复计算。
定义以下两个函数:
1.G(n):长度为 n 的序列能构造的不同的二叉搜索树的个数
2.F(i,n):以 i 为根、序列长度为 n 的不同的二叉搜索树的个数 (1 ≤ i ≤ n)。
因此,不同的二叉搜索树的总数 G(n),是对遍历所有 i(1 ≤ i ≤ n) 的 F(i, n)之和,即:
G(n)=
∑
i
=
1
N
F
(
i
,
n
)
\sum_{i=1}^NF(i,n)
∑i=1NF(i,n),对于边界情况,G(0)= G(1)= 1
给定序列 1 ~ n,我们选择数字 i 作为根,则根为 i 的所有二叉搜索树的集合是左子树集合和右子树集合的笛卡尔积,对于笛卡尔积中的每个元素,加上根节点之后形成完整的二叉搜索树
举个例子,创建以 3 为根、长度为 7 的不同二叉搜索树,整个序列是 [1, 2, 3, 4, 5, 6, 7],我们需要从左子序列 [1, 2] 构建左子树,从右子序列 [4, 5, 6, 7] 构建右子树,然后将它们组合(即笛卡尔积)。而G(n)的值与序列的内容无关,只与序列的长度有关,所以F(3)= G(2)* G(4)。
因此,可以推导出公式:F(i,n)= G(i - 1)* G(n - i)
G(i)=
∑
i
=
1
N
G
(
i
−
1
)
∗
G
(
n
−
i
)
\sum_{i=1}^NG(i - 1) * G(n - i)
∑i=1NG(i−1)∗G(n−i)
代码
class Solution {
public int numTrees(int n) {
int[] G = new int[n + 1];
G[0] = 1;
G[1] = 1;
for (int i = 2; i <= n; ++i) {
for (int j = 1; j <= i; ++j) {
G[i] += G[j - 1] * G[i - j];
}
}
return G[n];
}
}
时间复杂度为O(n2),空间复杂度为O(n)