不同的二叉查找树的形态

"""
设计：Python程序设计
作者：初学者
日期：2022年 05月 01日
"""


# 例115   不同的二叉查找树
#     1.问题描述
#     给定正整数n，求以1~n为节点组成的不同的二叉查找树有多少种？
#     2.问题示例
#     输入n=3，输出5，表示有5种不同形态的二叉查找树。
#     3.代码实现
class Solution:
    """
    参数n：整数
    返回值：整数
    """

    def num_trees(self, n):
        ans = [1, 1, 2]
        if n <= 2:
            return ans[n]
        else:
            ans += [0 for i in range(n - 2)]

            for i in range(3, n + 1):
                for j in range(1, i + 1):
                    ans[i] += ans[j - 1] * ans[i - j]
            return ans[n]


# 主函数
if __name__ == '__main__':
    s = Solution()
    n = 3
    print("输入：", n)
    print("输出：", s.num_trees(n))