leetcode 96. 不同的二叉搜索树

2023.8.10

本题可以先在草稿上画出各种不同的二叉搜索树,从中寻找规律,找出递推关系:

例如:当n=3时,分三种情况:

①1作为根节点,此时左子树有0个节点,右子树有2两个节点,此时的二叉树种数应该有dp[0]*dp[2]种。

②2作为根节点,此时左右子树各有一个节点,种数都是dp[1],此时的二叉树种数应该有dp[1]*dp[1]种。

③3作为根节点,此时情况和第一种情况类似,只是左右子树交换了。

于是可以分析得出:dp[i] 需要从1到 i 进行 i 次累加,每次累加的内容为:dp[左子树的节点个数] * dp[右子树的节点个数]。 代码的步骤如下:

  • 定义dp数组,大小为n。 须明确dp[i] 的含义:代表由i个节点组成的二叉搜索树的种数。
  • 初始化dp数组,当i=0 或者i=1时,此时二叉搜索树的种数都为1。
  • 用两层for循环对dp[i]进行更新。

具体代码如下:

cpp 复制代码
class Solution {
public:
    int numTrees(int n) {
        vector<int> dp(n+1); //dp[i]代表:由i个节点组成的二叉搜索树的种数。
        dp[0] = 1;
        dp[1] = 1;
        for(int i=2; i<=n; i++)
        {
            for(int j=1; j<=i; j++)
            {
                dp[i] += dp[j-1] * dp[i-j]; //左子树有j-1个节点,右子树有i-j个节点。
            }
        }
        return dp[n];
    }
};

很不错的题目,日后二刷。

相关推荐
美团技术团队22 分钟前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
River4162 小时前
Javer 学 c++(十三):引用篇
c++·后端
感哥4 小时前
C++ std::set
c++
Fanxt_Ja5 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下5 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶5 小时前
算法 --- 字符串
算法
博笙困了5 小时前
AcWing学习——差分
c++·算法
NAGNIP5 小时前
认识 Unsloth 框架:大模型高效微调的利器
算法
NAGNIP5 小时前
大模型微调框架之LLaMA Factory
算法
echoarts5 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust