【LeetCode】96. 不同的二叉搜索树（C++）

最新推荐文章于 2024-09-19 19:39:54 发布

m0_70234528

最新推荐文章于 2024-09-19 19:39:54 发布

阅读量135

点赞数

分类专栏： LeetCode 文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/weixin_53235104/article/details/130892269

版权

LeetCode 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

该问题要求计算由n个不同节点值构成的二叉搜索树的种类数。采用动态规划方法，定义dp[i]为含有i个节点的二叉搜索树的数量。通过递推公式dp[i]=Σ(dp[j]*dp[i-1-j])，其中j表示左子树的节点数，可以计算出所有可能的组合。程序使用Java实现，初始化dp[0]=dp[1]=1，然后遍历从2到n，计算每个dp[i]的值。

摘要由CSDN通过智能技术生成

题目描述：

给你一个整数 n ，求恰由 n 个节点组成且节点值从 1 到 n 互不相同的二叉搜索树有多少种？返回满足题意的二叉搜索树的种数。

示例 1：

输入：n = 3
输出：5

示例 2：

输入：n = 1
输出：1

提示：

1 <= n <= 19

思路：动态规划

dp[i] 的定义为：由 n 个节点组成的互不相同的二叉搜索树的个数
找规律：以 n = 3 为例，由 3 个节点组成的二叉搜索树个数 = 以 1 为头结点搜索树的数量 + 以 2 为头结点搜索树的数量 + 以 3 为头结点搜索树的数量，此时左子树和右子树的节点数分别为 0 和 2、1 和 1 及 2 和 0，即 dp[3] = dp[0] * dp[2] + dp[1] * dp[1] + dp[2] * dp[0];
确定递推公式：同理，对于 n 个节点组成的二叉搜索树，需要将头节点分别为 1 - n 的二叉搜索树个数累加，由此得到递推公式：dp[i] += dp[j] * dp[i-1-j]; 其中 j 表示左子树的节点数，i-1-j 表示右子树的节点数
数组的初始化：dp[0] = 1; dp[1] = 1; 因为当节点数为 0 和 1 时都只有一种情况
遍历顺序：从前往后
打印验证，结果正确

    class Solution {
    public:
        int numTrees(int n) {
            vector<long int> dp(n + 1);
            dp[0] = 1;
            dp[1] = 1;
            for (int i = 2; i <= n; i++) {
                // 有j个节点的左子树
                for (int j = 0; j < i; j++) {
                    dp[i] += dp[j] * dp[i-1-j];
                }
            }
            return dp[n];
        }
    };