[Leetcode]Unique Binary Search Trees II

Given n, generate all structurally unique BST's (binary search trees) that store values 1...n.

For example,
Given n = 3, your program should return all 5 unique BST's shown below.

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

给出N个节点的二叉搜索树的所有形态~每次选取一个结点做为根,然后递归求解左右子树的所有形态~

注意点:helper函数里l > r 时,return [None] ,是用来表示空树,这样也可以保证下面循环时即使一边是空树,也会跑另一边

class Solution:
    # @return a list of tree node
    def generateTrees(self, n):
        return self.helper(1, n)
        
    def helper(self, l, r):
        if l > r:
            return [None]
        res = []
        for rootVal in xrange(l, r + 1):
            leftList = self.helper(l, rootVal - 1)
            rightList = self.helper(rootVal + 1, r)
            for i in leftList:
                for j in rightList:
                    root = TreeNode(rootVal)
                    root.left = i
                    root.right = j
                    res.append(root)
        return res


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值