动态规划——统计不同二查搜索树

1、题目：力扣原题

2、分析

根据题意，先进行简单的分析：

假如只有一个节点，那么只存在一棵搜索树；若存在两个节点，则存在两棵搜索树，如下图所示：

 当节点数为n=3个时，我们可以得到5棵树：

 当根节点为1时，存在两种，其右子树有两个节点，布局类似与n=2时的两棵树形状布局；

当根节点为2时，存在1中，左右子树各一个节点，布局类似于n=1；

当根节点为3时，存在两种，其左子树有两个节点，布局类似于n=2;

所以，节点数n=3时的搜索树总数 = 1为根节点搜索树目+2为根节点搜索树数目+3为根节点搜索树数目；

1为根节点搜索树目 = 右子树有2个节点搜索树数量*左子树有0个节点搜索树数量

2为根节点搜索树目 = 右子树有1个节点搜索树数量*左子树有1个节点搜索树数量

3为根节点搜索树目 = 右子树有0个节点搜索树数量*左子树有2个节点搜索树数量

（有2个元素的搜索树数量就是dp[2]，有1个元素的搜索树数量就是dp[1]，有0个元素的搜索树数量就是dp[0]。）

即可以得到n=3的计算公式：dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]

-----------------

根据以上分析，我们可以再次采用dp五步曲的前四步进行整理：

1）确定dp

        dp[i] 表示从1到节点i组成的二叉搜索树的数目；

2）递推公式

        dp[i] += dp[j-1]*dp[i-j]； 其中，j-1表示以j为头节点左子树的数量，i-j表示以j为头结点右子树的数量；

3）初始化

        初始化dp[0] = 1，空节点也是一棵二叉树，也是一棵二叉搜索树；

4）遍历顺序

        节点数为i的状态是依靠 i之前节点数的状态，所以遍历顺序是节点数的小到大遍历

3、代码

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];
    }
}