本题是要求以n个节点组成,且节点值为1-n互不相同的二叉搜索树有多少种。我们可以采用动态规划的方式来做。
二叉搜索数是其左节点的值比根节点小,右节点的值比根节点大,并且其左右子树也都是二叉搜索树。
也就是说,我们需要从1-n轮流当根节点,然后判断其个数。
dp[i]我们定义为以1到i为根节点,其二叉搜索树的个数,假设此时根节点为j,那么其左子树的头节点最大只能是j-1,右子树应该是i-j。
也就是说 此时dp[i]=dp[j-1]*dp[i-j]。但是我们要把每次根节点的情况都累加起来,所以递推公式应该是dp[i]+=dp[j-1]*dp[i-j]。
然后初始化,dp[0]=1 dp[1]=1.dp[1]为1没什么好说的,根节点是1,只有一种情况。但是dp[0]=1,我们可以认为这是个空二叉搜索树,所以是1,也可以从遍历来判断,我们要的是乘积和,如果有一个元素是0,那么我们其中一部分的乘积就为0了,这就会导致结果丢失,所以要初始化为1.
遍历顺序:我们i要从小到大遍历到n,而j也是要从小到大遍历到i,因为我们是在i里面选的j,j最大也就只能是i-1;
打印数组
class Solution {
public int numTrees(int n) {
//初始化 dp 数组
int[] dp = new int[n + 1];
//初始化0个节点和1个节点的情况
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i <= n; i++) {
for (int j = 1; j <= i; j++) {
//对于第i个节点,需要考虑1作为根节点直到i作为根节点的情况,所以需要累加
//一共i个节点,对于根节点j时,左子树的节点个数为j-1,右子树的节点个数为i-j
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
}