题目
给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例1:
输入:n = 3
输出:5
示例 2:
输入:n = 1
输出:1
解题思路
本题使用动态规划的解题思路:
- dp[i] :使用一维数组即可。i表示有几个节点,0<i<=n,dp[i]表示有i个节点的二叉排序树有几种;
- 递推公式 :dp[i] += dp[j-1] * dp[i-j] 需要嵌套循环,dp[i]是一个累加的过程;
- 初始化 :dp[0] = 1, dp[1] = 1;
- 遍历顺序:从小到大遍历;
python代码
去博客设置页面,选择一款你喜欢的代码片高亮样式,下面展示同样高亮的 代码片.
class Solution:
def numTrees(self, n:int) ->int:
sum_bst = 0 # 存储有n个节点的二叉树共有几种树,累加
# 初始化dp数组
dp = [0]*(n+1)
dp[0], dp[1] = 1, 1
if n < 2:
return dp[n]
for i in range(2, n+1):
for j in range(1, i+1):
dp[i] += dp[j-1] * dp[i-j]
return dp[n]
if __name__ == "__main__":
n = 4
s = Solution()
count = s.numTrees(n)
print(count)
8741

被折叠的 条评论
为什么被折叠?



