给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
示例 2:输入:n = 1
输出:1
提示:
1 <= n <= 19
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public int numTrees(int n) {
int[] dp = new int[n+1];
dp[0] = 1;
dp[1] = 1;
if(n<3) return dp[n];
for(int i = 3;i<=n;i++){
for(int j = 1;j<=i;j++){
dp[i] += dp[j-1]*dp[i-j];
}
}
/*for(int i=0;i<=n;i++)
System.out.print(dp[i]+" ");*/
return dp[n];
}
}
这道题涉及到树的知识,我不是很懂,直接看了答案。。。
首先先分析n=1,n=2,n=3的情况,才可能递推出dp公式
n为1和n为2是比较直观的。
但是n=3开始就比较复杂了,
这时我们将其按照头节点的数字划分
① 当头节点为 1 时,只有右子树,其中右子树的两种形状是不是和n=2的那两种搜索树相似?
② 当头节点为 2 时,左子树一种情况,右子树一种情况,是不是和n=1的那一颗搜索树一样?
③ 当头节点为 3 时,只有左子树两种情况,是不是和n=2那两种搜索树相似?
所以我们可以推出:
n=3的搜索树数量 = 当头节点为1时 + 当头节点为2时 + 当头节点为3时
头节点为 1 时 = 【n=0】* 【n=2】
头节点为 2 时 = 【n=1】* 【n=1】
头节点为 3 时 = 【n=2】* 【n=0】
可以推出:
dp[n] = dp[0]*dp[n-1] + dp[1]*dp[n-2] + dp[2]*dp[n-3] + .....+ dp[n-1]*dp[0]
dp数组的含义:节点为1~n的二叉搜素树总数量
dp数组的初始化只需要dp[0] = 1
dp数组的循环:
for(int i = 2;i<=n;i++){
for(int j = 1;j<=i;j++){
dp[i] += dp[j-1]*dp[i-j];
}
}
dp[j-1]代表左子树,dp[i-j]代表右子树
外层循环应该从左往右,因为后面的dp数值都是由前面的推出的。