96. 不同的二叉搜索树
记忆化递归
class Solution {
public:
int dfs(int n, vector<int>& dp){
if(dp[n] != 0){
return dp[n];
}
for(int i = 0; i < n; i++){
//i代表的是左子树中含有结点的个数
//n - 1代表的是除去根节点外还剩多少个结点
//n - 1 - i自然代表的就是除去左子树中含有的结点和根节点之外剩余的节点的个数
//也就是右子树含有的结点的个数
dp[n] += dfs(i, dp) * dfs(n - 1 - i, dp);
}
return dp[n];
}
int numTrees(int n) {
vector<int> dp(n + 1);
//空树的情况
dp[0] = 1;
//一个节点的情况
dp[1] = 1;
return dfs(n, dp);
}
};
动态规划1
递推公式为:dp[n] += dp[j] * dp[i - 1 - j];
含义:j直接代表左子树含有的结点个数,i - 1 - j代表右子树含有的结点个数(具体为什么看上一段代码块里的注释)
class Solution {
public:
int numTrees(int n) {
if(n <= 2){
return n;
}
vector<int> dp(n + 1);
dp[0] = 1;
for(int i = 1; i <= n; i++){
//内循环j从0开始
for(int j = 0; j < i; j++){
dp[i] += dp[j] * dp[i - 1 - j];
}
}
return dp[n];
}
};
动态规划2
递推公式为:dp[i] += dp[j - 1] * dp[i - j];
含义:j相当于是头结点的元素,从1遍历到i为止
j-1 为j为头结点左子树节点数量,i-j 为以j为头结点右子树节点数量
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n + 1);
dp[0] = 1;
for (int i = 1; i <= n; i++) {
//内循环j从1开始
for (int j = 1; j <= i; j++) {
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
};