给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
从1到n将每个数看成root:
class Solution:
def numTrees(self, n):
"""
:type n: int
:rtype: int
"""
# 假设n个节点存在二叉排序树的个数是G(n),1为根节点,2为根节点,...,n为根节点,
# 当1为根节点时,其左子树节点个数为0,右子树节点个数为n-1,同理当2为根节点时,
# 其左子树节点个数为1,右子树节点为n-2,所以可得
# G(n) = G(0)*G(n-1)+G(1)*(n-2)+...+G(n-1)*G(0)
res = [0] * (n+1)
res[0] = 1
for i in range(1, n+1):
for j in range(i):
res[i] += res[j] * res[i-j-1]
return res