96. 不同的二叉搜索树
给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
示例 2:
输入:n = 1
输出:1
提示:
1 <= n <= 19
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
利用动态规划,找出状态转移方程。
设 g(n) 表示 n 个节点的不同二叉搜索树个数,设 f(i,n) 是以 i 为根、序列长度为 n 的不同二叉搜索树个数,那么有 g(n) = f(1, n) + f(2, n) + … + f(n, n)。边界条件 g(0) = 1, g(1) = 1,对于 f(i,n),应该是以 i 为根节点的不同左子树和不同右子树的积,即 f(i,n) = g(i - 1) * g(n - i),那么 g(n) = f(1, n) + f(2, n) + … + f(n, n) 中的 f 全部可以用 g 来代替。
题解(Java)
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];
}
}