★[leetCode 96] 不同的二叉搜索树

最新推荐文章于 2024-10-01 05:04:32 发布

不吃冰的北极熊

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量45

点赞数

分类专栏： leetCode刷题文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/weixin_43810348/article/details/125217416

版权

leetCode刷题专栏收录该内容

71 篇文章 0 订阅

订阅专栏

解题思路

也是动态规划，这题也是难在没办法确定递推公式，怎么知道dp[i]和之前的关系？

是这样的，我们在确定二叉搜索树的时候，不用管它每个节点是上放了什么数，只用管它的样子就行了，举个例子，假设二叉搜索树有3个节点，我们可以发现它每一个样子只会有一种树，不会有样子相同，但是节点的数不一样的二叉搜索树（指的是在题目条件下，没有重复的数）。

确定了这个，我们就可以计算当前有多少种二叉搜索树了。还是以3个节点为例，可以分成3种情况：

左子树0个节点，右子树2个节点；
左子树1个节点，右子树1个节点；
左子树2个节点，右子树0个节点。

那么动态规划的递推公式也就有了。
dp[i]=sum{ dp[j]*dp[i-j-1] } ，0≤j≤i-1，之所以要-1是因为还有一个根节点。

代码

class Solution {
    public int numTrees(int n) {
        int[] dp=new int[n+1];
        dp[0]=1;
        dp[1]=1;
        for (int i=2;i<=n;i++){
            int sum=0;
            for (int j=0;j<i;j++){
                sum=sum+dp[j]*dp[i-j-1];
            }
            dp[i]=sum;
        }
        return dp[n];
    }
}