给你一个整数 n
,求恰由 n
个节点组成且节点值从 1
到 n
互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3 输出:5
示例 2:
输入:n = 1 输出:1
二叉搜索树又称为二叉排序树,它或者是一棵空树,或者是具有以下性质的二叉树:
- 若它的左子树不为空,则左子树上所有结点的值都小于根结点的值。
- 若它的右子树不为空,则右子树上所有结点的值都大于根结点的值。
- 它的左右子树也分别是二叉搜索树。
首先来确定本题的dp[i]是什么意思,应该是输出的二叉搜索树的数量
例如dp[0]=dp[1] =1,在没有节点和只有一个节点时,二叉搜索树只会出现一种情况
dp[2]时,会出现两种情况,头节点有左孩子或者有右孩子
dp[3],就是 元素1为头结点搜索树的数量 + 元素2为头结点搜索树的数量 + 元素3为头结点搜索树的数量
元素1为头结点搜索树的数量 = 右子树有2个元素的搜索树数量 * 左子树有0个元素的搜索树数量
元素2为头结点搜索树的数量 = 右子树有1个元素的搜索树数量 * 左子树有1个元素的搜索树数量
元素3为头结点搜索树的数量 = 右子树有0个元素的搜索树数量 * 左子树有2个元素的搜索树数量
所以dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]
class Solution:
def numTrees(self, n: int) -> int:
dp = [0]*(n+1)
dp[0]=dp[1]=1
for i in range(2,n+1):#i在2到i的范围内
for j in range(1,i+1):#j是头节点
dp[i]+=dp[j-1]*dp[i-j]#头节点的左子树的节点数量一定是j-1,右子树是i-j
return dp[n]