【Leetcode hot 100】96. 不同的二叉搜索树

题目链接

思路

动态规划
题目是给定一个有序序列1,2,3…n, 我们需要求出一共有多少种构建二叉搜索树的方法。可以这样考虑：对每一个数字i，我们遍历它，以i为根节点，1…(i-1)为左子树，i+1…n为右子树，接着我们可以以同样的方式对左右子树递归构建二叉树。
上述遍历中由于根节点不同，所以每次构建的二叉树都不相同。由此可见，原问题可以分解成规模较小的两个子问题，且子问题的解可以复用。因此，我们可以想到使用动态规划来求解本题。

假设两个函数：
G(n):长度为n的序列构成的不同二叉搜索树的个数。
F(i,n):以i为根节点，序列长度为n构成的二叉树的个数。
可见，G(n)G(n)G(n) 是我们求解需要的函数。

稍后我们将看到，G(n)G(n)G(n) 可以从 F(i,n)F(i, n)F(i,n) 得到，而 F(i,n)F(i, n)F(i,n) 又会递归地依赖于 G(n)G(n)G(n)。

首先，根据上一节中的思路，不同的二叉搜索树的总数 G(n)G(n)G(n)，是对遍历所有 iii (1≤i≤n)(1 \le i \le n)(1≤i≤n) 的 F(i,n)F(i, n)F(i,n) 之和。换言之：

G(n)=∑i=1nF(i,n)(1) G(n) = \sum_{i=1}^{n} F(i, n)\qquad \qquad (1)
G(n)=
i=1
∑
n
​
F(i,n)(1)
对于边界情况，当序列长度为 111（只有根）或为 000（空树）时，只有一种情况，即：

G(0)=1,G(1)=1 G(0) = 1, \qquad G(1) = 1
G(0)=1,G(1)=1
给定序列 1⋯n1 \cdots n1⋯n，我们选择数字 iii 作为根，则根为 iii 的所有二叉搜索树的集合是左子树集合和右子树集合的笛卡尔积，对于笛卡尔积中的每个元素，加上根节点之后形成完整的二叉搜索树，如下图所示：

举例而言，创建以 333 为根、长度为 777 的不同二叉搜索树，整个序列是 [1,2,3,4,5,6,7][1, 2, 3, 4, 5, 6, 7][1,2,3,4,5,6,7]，我们需要从左子序列 [1,2][1, 2][1,2] 构建左子树，从右子序列 [4,5,6,7][4, 5, 6, 7][4,5,6,7] 构建右子树，然后将它们组合（即笛卡尔积）。

对于这个例子，不同二叉搜索树的个数为 F(3,7)F(3, 7)F(3,7)。我们将 [1,2][1,2][1,2] 构建不同左子树的数量表示为 G(2)G(2)G(2), 从 [4,5,6,7][4, 5, 6, 7][4,5,6,7] 构建不同右子树的数量表示为 G(4)G(4)G(4)，注意到 G(n)G(n)G(n) 和序列的内容无关，只和序列的长度有关。于是，F(3,7)=G(2)⋅G(4)F(3,7) = G(2) \cdot G(4)F(3,7)=G(2)⋅G(4)。 因此，我们可以得到以下公式：

F(i,n)=G(i−1)⋅G(n−i)(2) F(i, n) = G(i-1) \cdot G(n-i) \qquad \qquad (2)
F(i,n)=G(i−1)⋅G(n−i)(2)
将公式 (1)(1)(1)，(2)(2)(2) 结合，可以得到 G(n)G(n)G(n) 的递归表达式：

G(n)=∑i=1nG(i−1)⋅G(n−i)(3) G(n) = \sum_{i=1}^{n}G(i-1) \cdot G(n-i) \qquad \qquad (3)
G(n)=
i=1
∑
n
​
G(i−1)⋅G(n−i)(3)
至此，我们从小到大计算 GGG 函数即可，因为 G(n)G(n)G(n) 的值依赖于 G(0)⋯G(n−1)G(0) \cdots G(n-1)G(0)⋯G(n−1)。

代码：

class Solution {
public:
    int numTrees(int n) {
        vector<int> G(n + 1, 0);
        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];
    }
};