LeetCode_96. 不同的二叉搜索树（Java）

题目描述

 

原题地址

---

思路

 
由于每一个序列节点都有可能成为根节点，所以在遍历每个序列节点时，显然会出现重复计算左右子树的种类个数的情况，因此可以想到用DP来解决，避免大量的重复计算。
定义以下两个函数：
1.G（n）：长度为 n 的序列能构造的不同的二叉搜索树的个数
2.F（i，n）：以 i 为根、序列长度为 n 的不同的二叉搜索树的个数 (1 ≤ i ≤ n)。
因此，不同的二叉搜索树的总数 G(n)，是对遍历所有 i(1 ≤ i ≤ n) 的 F(i, n）之和，即：
G（n）= ${\sum }_{i=1}^{N}F(i,n)$，对于边界情况，G（0）= G（1）= 1
给定序列 1 ~ n，我们选择数字 i 作为根，则根为 i 的所有二叉搜索树的集合是左子树集合和右子树集合的笛卡尔积，对于笛卡尔积中的每个元素，加上根节点之后形成完整的二叉搜索树
举个例子，创建以 3 为根、长度为 7 的不同二叉搜索树，整个序列是 [1, 2, 3, 4, 5, 6, 7]，我们需要从左子序列 [1, 2] 构建左子树，从右子序列 [4, 5, 6, 7] 构建右子树，然后将它们组合（即笛卡尔积）。而G（n）的值与序列的内容无关，只与序列的长度有关，所以F（3）= G（2）* G（4）。
因此，可以推导出公式：F（i，n）= G（i - 1）* G（n - i）
G（i）= ${\sum }_{i=1}^{N}G(i-1)\ast G（n-i）$

---

代码

class Solution {
    public int numTrees(int n) {
        int[] G = new int[n + 1];
        G[0] = 1;
        G[1] = 1;

        for (int i = 2; i <= n; ++i) {
            for (int j = 1; j <= i; ++j) {
                G[i] += G[j - 1] * G[i - j];
            }
        }
        return G[n];
    }
}

时间复杂度为O（n²），空间复杂度为O（n）