力扣96.不同的二叉搜索树

题目来源:96.不同的二叉搜索树
给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
力扣96.不同的二叉搜索树_第1张图片
思路:动态规划
给定有序数列1:n,当搜索子树的数量时,假设以i为根的二叉搜索树个数为f(i);给定n个结点能构成的二叉搜索数个数为G(n);则
G(n)=f(0)+f(1)+f(2)+…+f(n)
分别代表【以第1个结点为根结点:前0个结点在左,后n-1个结点在右计算子树个数;以第2个结点为根结点:前1个结点在左,后n-2个结点在右计算子树个数;以第3个结点为根结点:前2个结点在左,后n-3个结点在右计算子树个数;…; 以第n个结点为根结点:前n-1个结点在左,后0个结点在右计算子树个数】一共n+1中情况。
进一步,我们来讨论一下f(i)的计算。f(i)表示:前i-1个结点在左,后n-i个结点在右的子树个数。而前i-1个结点在左的搜索树个数为:G(i-1),后n-i个结点的搜索树个数为G(n-i),则有:
f(i)=G(i-1)*G(n-i)
最后将f(i)的公式代入,即求得G(n)的表达式:
G(n)=G(0)*G(n-1)+G(1)*G(n-1)+…+G(n)*G(0)
最后代码如下:

class Solution(object):
    def numTrees(self, n):
        """
        :type n: int
        :rtype: int
        """
        f = [0 for _ in range(n+1)]
        f[1] = 1
        f[0] = 1
        for i in range(2,n+1):
        	for j in range(i+1):
        		f[i] += f[i-1]*f[i-j]
        return f[-1]

代码运行结果如下:
力扣96.不同的二叉搜索树_第2张图片
另一种解法如下:

class Solution(object):
    def numTrees(self, n):
        """
        :type n: int
        :rtype: int
        """
        f = [0 for _ in range(n+1)]
        f[1] = 1
        f[0] = 1
        for j in range(2, n+1):
            for i in range(0, n):
                f[j] += f[i]*f[j-i-1]
        return f[-1]

运行结果如下:
力扣96.不同的二叉搜索树_第3张图片
可以看出两种方法差不多,但是空间和时间完全相反,还没理解两者的不同,有理解的小伙伴可以一起讨论一下~
方法一参考链接
方法二参考链接

你可能感兴趣的:(力扣刷题笔记,leetcode,算法,职场和发展)