[Leetcode][python]Unique Binary Search Trees/不同的二叉查找树

【Leetcode题解】 同时被 2 个专栏收录
146 篇文章 0 订阅

题目大意

给出一个n,求1-n能够得到的所有二叉搜索树

解题思路

转自博客

这题想了好久才想清楚。其实如果把上例的顺序改一下,就可以看出规律了。
这里写图片描述

比如,以1为根的树有几个,完全取决于有二个元素的子树有几种。同理,2为根的子树取决于一个元素的子树有几个。以3为根的情况,则与1相同。

定义Count[i] 为以[0,i]能产生的Unique Binary Tree的数目,

如果数组为空,毫无疑问,只有一种BST,即空树,Count[0] =1

如果数组仅有一个元素{1},只有一种BST,单个节点Count[1] = 1

如果数组有两个元素{1,2}, 那么有如下两种可能

1                       2
  \                    /
    2                1

Count[2] = Count[0] * Count[1] (1为根的情况)
+ Count[1] * Count[0] (2为根的情况。

再看一遍三个元素的数组,可以发现BST的取值方式如下:

Count[3] = Count[0]*Count[2] (1为根的情况)
+ Count[1]*Count[1] (2为根的情况)
+ Count[2]*Count[0] (3为根的情况)

所以,由此观察,可以得出Count的递推Count[i] = ∑ Count[0…k] * [ k+1….i] 0<=k

代码

其实也就是求卡特兰数的定义实现
这里写图片描述

class Solution(object):
    def numTrees(self, n):
        """
        :type n: int
        :rtype: int
        """
        dp = [1, 1, 2]
        if n <= 2:
            return dp[n]
        else:
            dp += [0 for i in range(n-2)]  # 后面创建多个
            for i in range(3, n + 1):
                for j in range(1, i+1):
                    # 若n为4:dp[0]*dp[3]+dp[1]*dp[2]+dp[2]*dp[1]+dp[3]*dp[0]
                    dp[i] += dp[j-1] * dp[i-j]
            return dp[n]

总结

基于以下原则的BST建树具有唯一性:
以i为根节点的树,其左子树由[0, i-1]构成, 其右子树由[i+1, n]构成。

  • 1
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值