leetcode 96. Unique Binary Search Trees 递归法和动态规划法--python代码版

原题

首先,这道题并不是求简单的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]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值