题目描述:
标签:树 动态规划
给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
代码:
思路分析:动态规划五部曲
1、确定dp数组以及下标的含义——这里dp[i]是指1...i为节点组成的二叉搜索树个数
2、确定递推公式,其实举例n=1/2/3的情况就能找到规律了。
当n=0或1,只有1种情况dp[0]=1,dp[1]=1,
当n=2,有两种情况dp[2]=2,
当n=3,以1为根节点,左子树0个右子树2个,dp[0]*dp[2];以2为根节点,左子树1个右子树1个,dp[1]*dp[1];以2为根节点,左子树2个右子树0个,dp[2]*dp[0],
所以dp[3]=dp[0]*dp[2]+dp[1]*dp[1]+dp[2]*dp[0]。
所以dp[i] += dp[j-1]*dp[i-j]
3、dp数组初始化,dp[0]=1
4、确定遍历顺序,双层for循环遍历,i从1到n,j从1到i
5、举例推导dp数组
class Solution {
public int numTrees(int n) {
int[] dp = new int[n+1];
dp[0] = 1;
for(int i = 1;i <=n;i++){
for(int j = 1;j <= i;j++){
dp[i] += dp[j-1]*dp[i-j];
}
}
return dp[n];
}
}