给你一个整数 n
,求恰由 n
个节点组成且节点值从 1
到 n
互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3 输出:5
示例 2:
输入:n = 1 输出:1
提示:
1 <= n <= 19
class Solution {
public int numTrees(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 1;
if(n >= 2) dp[2] = 2;
for(int i = 3;i < n + 1;i++){
for(int j = 1;j <= i;j++){
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
}
题目分析: 当n为3时的5种情况分别为1,2,3为根节点,其中1为根节点时右子树是n为2时的二叉搜索树的两种情况,2为根节点时左右子树均是n为1时的二叉搜索树,3为根节点时右子树是n为2时的二叉搜索树的两种情况。因此 有3个节点的二叉搜索树的种数 = 头节点为1时的种数 + 头节点为2时的种数 + 头节点为3时的种数;而对于一个二叉搜索树而言,其种数为 其左子树的种数 * 右子树种数。
动态规划五部曲:
1.dp数组以及下标的含义:dp数组是不同的二叉搜索树的种数数组,其中dp[i]表示n为i时有dp[i]个不同的二叉搜索树
2.递推公式:dp[i] += dp[j - 1] * dp[i - j],其中1 <= j <= i
3.dp数组如何初始化:dp[0] = 1,dp[1] = 1,dp[2] = 2(如果n >= 2)
4.遍历顺序:从小到大
5.打印dp数组:用于debug自查代码