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];
    }
};

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

相关推荐
小张会进步1 分钟前
数组:二维数组
java·javascript·算法
佑白雪乐20 分钟前
LCR 175. 计算二叉树的深度
算法·深度优先
阿Y加油吧27 分钟前
力扣打卡day07——最大子数组和、合并区间
算法
Larry_Yanan31 分钟前
Qt网络开发之基于 QWebEngine 实现简易内嵌浏览器
linux·开发语言·网络·c++·笔记·qt·学习
想吃火锅100531 分钟前
【leetcode】105. 从前序与中序遍历序列构造二叉树
算法·leetcode·职场和发展
圣保罗的大教堂33 分钟前
leetcode 3567. 子矩阵的最小绝对差 中等
leetcode
2401_8318249637 分钟前
嵌入式C++驱动开发
开发语言·c++·算法
靠沿40 分钟前
【优选算法】专题十八——BFS解决拓扑排序问题
算法·宽度优先
cui_ruicheng43 分钟前
C++数据结构进阶:哈希表实现
数据结构·c++·算法·哈希算法·散列表
前端摸鱼匠43 分钟前
面试题4:多头注意力(MHA)相比单头注意力的优势是什么?Head数如何影响模型?
人工智能·ai·面试·职场和发展·求职招聘