• 【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];
        }
    };
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
  • 相关阅读:
    selenium也能过某数、5s盾..
    微服务使用指南
    Pytorch神经网络设计技巧-如何得到各层的input shape
    EventLoop
    OpenCV环境搭建
    视频剪辑大师:批量快进慢放,让你的视频瞬间生动起来!
    tomcat映射本地文件路径
    计算机网络-运输层详解(持续更新中)
    Spring基础(四):XML方式实现DI
    uni-app结合laravel实现免登陆
  • 原文地址:https://blog.csdn.net/weixin_43833610/article/details/130856397