95. Unique Binary Search Trees II

最新推荐文章于 2020-03-06 09:23:21 发布

原创最新推荐文章于 2020-03-06 09:23:21 发布 · 275 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

刷题专栏收录该内容

76 篇文章

订阅专栏

本文介绍了一种算法，用于生成所有可能的不同形态的二叉查找树，当输入整数n时，生成所有由1到n为节点组成的二叉查找树。通过递归方式构造左右子树，并组合成完整的树结构。

题目描述：

给定一个整数 n，生成所有由 1...n 为节点组成的不同的二叉查找树。

例如，
给定 n = 3，则返回所有 5 种不同形态的二叉查找树：

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

我的思路：

不会写，Discuss里面找了个代码改成了python版本的学习下。

我的代码：

Discuss：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def generateTrees(self, n):
        """
        :type n: int
        :rtype: List[TreeNode]
        """
        if n == 0:
            return []
        return self.generate(1, n)
    
    def generate(self, start, end):
        rel = []
        if start > end:
            rel.append(None)
            return rel
        
        for i in range(start, end + 1):
            leftSubtree, rightSubtree = [], []
            leftSubtree = self.generate(start, i - 1)
            rightSubtree = self.generate(i + 1, end)
            
            for left in leftSubtree:
                for right in rightSubtree:
                    root = TreeNode(i)
                    root.left = left
                    root.right = right
                    rel.append(root)
                    
        return rel

学到：