首页 > 其他分享 >【LeetCode动态规划#04】不同的二叉搜索树(找规律,有点像智力题)

【LeetCode动态规划#04】不同的二叉搜索树(找规律,有点像智力题)

时间:2023-03-25 23:14:28浏览次数:51  
标签:智力题 04 二叉 树有 搜索 为头 节点 LeetCode dp

不同的二叉搜索树

力扣题目链接(opens new window)

给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?

示例:

img

思路

题意分析

先找一下关系

当n = 1时,如果元素就是1,以1为头节点

1

当n = 2时,分别以1和2为头节点

  1     2
 /       \
2         1 

然后当n = 3时的情况就是示例中给的那几种

找找有什么规律

当n = 3且使用1为头节点时,其子树的布局和n = 2时的布局是一样的(注意看1-2、2-1和3-2、2-3的方向,是不是一样的,数值不同没有影响)

当n = 3且使用2为头节点时,其左右子树布局和n = 1时的布局是一样的(n = 1是左右子树为空,也算1种情况)

当n = 3且使用3为头节点时,其子树的布局和n = 2时的布局是一样的

某种程度上,n = 3的二叉搜索树种类情况可以由n = 2以及n = 1推导出来

因此,n = 3时,二叉搜索树种类 = 头节点为1时的情况+头节点为2时的情况+头节点为3时的情况,来组成

即,头1+头2+头3

公式描述

接下来分析不同头节点时的情况

从图中可以看出,头节点为1时有:

 1       1
  \       \
   3       2
  /         \
 2           3

头节点为1时有多少种二叉搜索树可以用以下公式描述:

头1 = 左子树有0个节点时有几种二叉搜索树 * 右子树有2个节点时有几种二叉搜索树;(2种)

如何理解?

头节点为1来构建二叉搜索树的话,如果左子树只给0个节点,那么左子树的类型就只有1种(也就是空);然后给右子树2个节点的话,那么右子树就可以有3-2和2-3两种情况。

左右子树的情况组合在一起就得到以下结论:

​ 当n = 3时,使用1作为头节点可以构建2种(1*2)不同的二叉搜索树

还不理解再举个例子:

       10
      /  \
 5个节点  10个节点 

上述以10(数值无所谓)为头节点的二叉树,其左右子节点的情况如上

那么,可以构成的二叉树的种类一共是:5*10种

同理可以得到头2、头3的公式描述:

头2 = 左子树有1个节点时有几种二叉搜索树 * 右子树有1个节点时有几种二叉搜索树;(1种)

头3 = 左子树有2个节点时有几种二叉搜索树 * 右子树有0个节点时有几种二叉搜索树;(2种)

与示例对照的话发现是可以对上的

还是先来五部曲吧

五步走

1、确定dp数组的含义

根据题目所求可以得到

dp[i]:输入为i时有dp[i]种不同的二叉搜索树

2、确定递推公式

由前面的分析可以知道,当输入n为3时,可以组成的二叉搜索树种类(也就是dp[3])是有分别使用1、2、3作为头节点时产生的种类相加得到的。

即,

dp[3] = 头1+头2+头3
dp[3] = dp[0]dp[2] + dp[1]dp[1] + dp[2]dp[0];

明确了上述问题后可以开始讨论dp[i]

dp[i]可以从哪里求出来?

那肯定是由以1、2、3...i为头节点的所有情况相加得出,于是我们需要枚举所有头节点情况

j 来代表头节点数,那么该二叉搜索树的左子树有多少个节点?答案是 j-1

(举个例子来理解:示例中以3为头节点时,其左子树是不是有两个节点)

那么该二叉搜索树的右子树有多少个节点?答案是 i-j

因为我们这里是二叉搜索树,现在以i为头节点了,右子树的节点值一定都比 i 大,总节点数是 i ,那么留给右子树的节点数就是i-j了

套用dp数组的定义:

输入为 j-1 时有 dp[j-1] 种不同的二叉搜索树;

输入为 i-j 时有 dp[i-j] 种不同的二叉搜索树;

那么dp[i]怎么求?

根据 公式描述 中的讨论可得:dp[i] = dp[j-1] * dp[i-j];(只是当前j下的dp[i])

因为 j 是代表遍历所有头节点的情况,所以要把头节点为1~i的情况都相加才能得出最后的dp[i]

即递推公式应该是:dp[i] += dp[j-1] * dp[i-j];(i个节点有多少种不同的二叉搜索树)

怎么理解?

拿前面的例子dp[3]来说

dp[3] = dp[0]dp[2] + dp[1]dp[1] + dp[2]dp[0];

此处j的遍历范围是1~i,可以写成以下形式

dp[3] = dp[1-1]dp[3-1] + dp[2-1]dp[3-2] + dp[3-1]dp[3-3];

3、确定初始化方式

前面的讨论也说了dp[0]、dp[1] (即,n=0、n=1)可以用于推出后续情况

那么就要对这两者进行初始化吗?其实只需要初始化dp[0]就行了,dp[1]也可以通过dp[0]推出

dp[0]的含义是什么?输入的i是0,0个节点有多少种不同的二叉搜索树呢?答案是1个,因为空二叉树也是一种二叉搜索树

所以dp[0] = 1;

并且这也符合递推公式的要求,因为如果有空节点,其种类如果是0,那么不论后面的其他子树有几种情况,结果都是0,就没有意义了,因此空节点的种类应该是1(如果左右子树都空的话,也就是1*1=1,递推还能继续进行下去)

4、确定遍历顺序

还是从dp[3]来看

dp[3] = dp[0]dp[2] + dp[1]dp[1] + dp[2]dp[0];

dp[3]都是由小于3的状态累加推导来的,所以就要从小到大遍历,才可以利用之前遍历的状态

for(int i = 1; i <= n; ++i){
    for(int j = 1; j <= i; ++j){
        dp[i] += dp[j-1] * dp[i-j];
    }
}

代码

class Solution {
public:
    int numTrees(int n) {
        //定义dp数组
        vector<int> dp(n + 1);
        //初始化dp数组
        dp[0] = 1;
        //dp[1] = 1;//不用初始化dp[1],否则按理来说dp[1]应该是1,手动初始化会使其变为2
        //遍历
        for(int i = 1; i <= n; ++i){//此处i确实要取3,所以有等于号
            for(int j = 1; j <= i; ++j){
                dp[i] += dp[j - 1] * dp[i - j];
            }
        }
        // cout << dp[1]<< endl;//打印dp数组debug
        return dp[n];//确实要返回n的dp而不是n-1
    }
};

ps:真难啊动态规划

标签:智力题,04,二叉,树有,搜索,为头,节点,LeetCode,dp
From: https://www.cnblogs.com/DAYceng/p/17255860.html

相关文章

  • LeetCode199.二叉树的右视图
    1.题目:给定一个二叉树的根节点root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。示例1:输入: [1,2,3,null,5,null,4]输出: [1,3,4]来源:力......
  • LeetCode 59. 螺旋矩阵 II
    这道题可以采用模拟法来实现。我们可以设置上下左右四个边界,然后模拟螺旋填充元素。具体来说,我们定义left、right、top、bottom四个变量代表当前需要填充的最左边、最右......
  • Leetcode 17.电话号码的字母组合 (模拟)
    题目链接在这里:电话号码的字母组合这道题主要学习的是哈希表的应用:可以用大括号来代表建立哈希表,以及子函数的实现:可以直接在主函数中定义子函数,将\(string\)拼成一个整个......
  • Leetcode 15 & 16 (双指针)
    都是比较经典的双指针问题,我们可以从中总结一些双指针的规律首先这两题如果en做的话就是\(O(n^{3})\)的算法,暴力去找。但是我们可以发现这三个值是满足一定约束的,所以......
  • CS61A_HW04
    Q6题目描述:Writeafunction has_path thattakesinatree t andastring phrase.Itreturns True ifthereisapaththatstartsfromtherootwherethe......
  • #yyds干货盘点# LeetCode程序员面试金典:两数之和
    题目:给定一个整数数组nums 和一个整数目标值target,请你在该数组中找出和为目标值target 的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答......
  • #yyds干货盘点# LeetCode面试题:螺旋矩阵 II
    1.简述:给你一个正整数 n,生成一个包含1到 n2 所有元素,且元素按顺时针顺序螺旋排列的 nxn正方形矩阵matrix。 示例1:输入:n=3输出:[[1,2,3],[8,9,4],[7,6,5]]示......
  • LeetCode60. 排列序列
    classSolution{public:intfac[10];voidinit(){fac[0]=1;fac[1]=1;for(inti=2;i<10;i++)fac[i]=fac[i-1]*i......
  • ubuntu18.04离线 安装jdk8环境
    Jdkoracle官方下载地址:https://www.oracle.com/technetwork/java/javase/downloads/jdk8-downloads-2133151.html解压tar-zxvfjdk-8u152-linux-x64.tar.gz习惯上会......
  • Python爬虫基础——04-流程控制语句
    2.8,输出-输入2.8.1输出:#普通输出print('江户川柯南')#格式化输出#爬虫用法---在scrapy框架的时候输出到excel文件mysqlredisage=18name='工藤新......