Leetcode 96. 不同的二叉搜索树【动态规划】

Leetcode 96. 不同的二叉搜索树

中等题

题目

给你一个整数 n ，求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种？返回满足题意的二叉搜索树的种数。

示例1

输入：n = 3
输出：5

示例2

输入：n = 1
输出：1

提示

• 1 <= n <= 19

解法

可以拆分大问题到小问题，在指定 i 个节点的情况下，对于 j 为根节点时，左右子树的情况来计算当前的二叉搜索树数目

• dp表示n个节点组成的二叉搜索树的数目 dp[0] = 1
• 以j为根的情况下，左子树有(j-1)个节点，右子树有(i-j)个节点，那么j为根时不同的二叉搜索树的数目为dp[j-1]*dp[i-j]。
• 遍历[1,2,3,…,i]所有节点为根的情况，将其相加可以得到dp[i]

class Solution:
    def numTrees(self, n: int) -> int:
        dp =  [1 for _ in range(n+1)]  
        for i in range(1, n+1):
            tmp = 0
            for j in range(1,i+1): # 遍历更新dp[i]
                tmp += dp[j-1]*dp[i-j]  
            dp[i] = tmp
        return dp[n]

• 时间复杂度 $O(n^2)$
• 空间复杂度 O(n)

数学解法

class Solution(object):
    def numTrees(self, n):
        """
        :type n: int
        :rtype: int
        """
        C = 1
        for i in range(0, n):
            C = C * 2*(2*i+1)/(i+2)
        return int(C)