首先,这道题并不是求简单的n个节点的树形到底有多少种。
关键之处必须理解二叉排序树根节点大于左子树节点,小于右子树节点。
我们假设n个节点的二叉排序树有f(n)种,树中节点的值为1~n,
如果某树的根节点是i,那么此时有多少种的可能呢?
根节点左边的节点值均小于根节点,所以有i-1个节点,那么左子树就有f(i-1)种可能,同理右子树就有f(n-i)种可能。这样的话,当根节点为 i 时的可能数有f(i-1)*f(n-i)
那么总的可能数就是f(0) * f(n-1) + f(1) * f(n-2) f(2) * f(n-3)+…+f(n-1) * f(0)
这样的话,便可以很容易的写出递归版的代码:
class Solution:
def numTrees(self, n: int) -> int:
if n==0 or n==1:
return 1
sum = 0
for i in range(1, n+1):
sum += (self.numTrees(n-i) * self.numTrees(i-1))
return sum
当然这样的效率不高,leetcode上也跑不通。
下面贴动态规划版:
class Solution:
def numTrees(self, n: int) -> int:
F = [1,1]
for i in range(n-1):
F.append(0)
for i in range(2, n+1):
for j in range(1, i+1):
F[i] += (self.numTrees(j-1) * self.numTrees(i-j))
return F[n]
tmd这样还是跑不通,??
明天再找解法。。。。
我来了,现在有时间来改这个代码。
看了看,发现n<13时都可以,但是大于13 时就会Time Limit Exceeded,难受。。。。
找了一个代码逻辑基本相同的跑了一跑,可以运行,还是没想通为什么。给大家贴一下可以正确运行的代码,希望有小伙伴能通知我上面代码的错误。
正确代码:
class Solution:
def numTrees(self, n):
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-1-j]
return res[n]