代码随想录算法训练营第41天| 96.不同的二叉搜索树

链接: 96.不同的二叉搜索树

96.不同的二叉搜索树

递推公式有点难想到，没有考虑到左子树数量和右子树数量相乘的关系。
这道题目需要利用二叉搜索树的特征并思考头节点数值和左右子树数量的关系。

class Solution {
    public int numTrees(int n) {
        if(n == 0 || n == 1) return 1;

        int[] dp = new int[n+1];
        //初始化0个节点的情况
        dp[0] = 1;

        for(int i = 1; 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];
    }
}