给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \
3 2 1 1 3 2
/ / \
2 1 2 3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution:
def numTrees(self, n: int) -> int:
#1、 求f(i, n) = G(i-1) * G(n - i)=>f(i, n)为以i为根节点,长度为n 其值为左和右子树二叉搜索树个数乘积
#2、 G(n)为长度为n拥有的二叉搜索数的个数
# G(n) = f(1, n)+ ··· + f(n, n)
# 会超时
# if (n == 0 or n == 1):
# return 1
# sums = 0
# for i in range(1, n+1):
# sums += self.numTrees(i-1) * self.numTrees(n - i)
# return sums
#动态规划,依次求G(0),G(1) ··· G(n)
#方程和上面差不多
G = [0]*(n+1)
G[0], G[1] = 1, 1
for i in range(2, n+1):
for j in range(1, i+1):
G[i] += G[j-1] * G[i-j]
return G[n]