96. 不同的二叉搜索树
给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
分析:
确定dp数组及下标:i个节点的二叉搜索树有 dp[i] 种
递推公式:这道题的递推公式是比较难的,以三个节点举例,该树的结构可以分为:元素1为头结点搜索树的数量 + 元素2为头结点搜索树的数量 + 元素3为头结点搜索树的数量。
其中 元素1为头结点搜索树的数量 为:右子树有2个元素的搜索树数量 * 左子树有0个元素的搜索树数量
元素2为头结点搜索树的数量 为:右子树有1个元素的搜索树数量 * 左子树有1个元素的搜索树数量
元素3为头结点搜索树的数量 为:右子树有0个元素的搜索树数量 * 左子树有2个元素的搜索树数量
所以 dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]
初始值:dp[0] = 1
遍历顺序:从小到大
推导:[1, 2, 5 ...]
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n+1, 0);
dp[0] = 1;
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= i; ++j) {
dp[i] += dp[i-j] * dp[j-1];
}
}
return dp[n];
}
};