【力扣】96. 不同的二叉搜索树
给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
示例 2:
输入:n = 1
输出:1
提示:
1 <= n <= 19
题解
-
确定 dp 数组以及下标的含义
dp[i] : 1到 i 为节点组成的二叉搜索树的个数为 dp[i]
i 个不同元素节点组成的二叉搜索树的个数为 dp[i] -
确定递推公式
dp[3],就是 元素1为头结点搜索树的数量 + 元素2为头结点搜索树的数量 + 元素3为头结点搜索树的数量
元素1为头结点搜索树的数量 = 右子树有2个元素的搜索树数量 * 左子树有0个元素的搜索树数量
元素2为头结点搜索树的数量 = 右子树有1个元素的搜索树数量 * 左子树有1个元素的搜索树数量
元素3为头结点搜索树的数量 = 右子树有0个元素的搜索树数量 * 左子树有2个元素的搜索树数量
有2个元素的搜索树数量就是 dp[2]。有1个元素的搜索树数量就是 dp[1]。
所以dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]
dp[i] += dp[以 1到i-1 为头结点左子树节点数量] * dp[以 i-(1到i-1) 为头结点右子树节点数量]
文章来源:https://www.toymoban.com/news/detail-690392.html
所以递推公式:dp[i] += dp[j - 1] * dp[i - j]
; ,j-1 为 j 为头结点左子树节点数量,i-j 为以 j 为头结点右子树节点数量。文章来源地址https://www.toymoban.com/news/detail-690392.html
-
dp 数组如何初始化
dp[0] = 1 -
确定遍历顺序
节点数为 i 的状态是依靠 i 之前节点数的状态。
那么遍历 i 里面每一个数作为头结点的状态,用 j 来遍历
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= i; j++) {
dp[i] += dp[j - 1] * dp[i - j];
}
}
- 举例推导 dp 数组(打印 dp 数组)
class Solution {
public int numTrees(int n) {
//初始化
int[] dp = new int[n + 1];
//初始化0个节点和1个节点的情况
dp[0] = 1;
// 遍历
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= i; j++) {
// dp 方程
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
}
到了这里,关于【力扣】96. 不同的二叉搜索树 <动态规划>的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!