LeetCode Unique Binary Search Trees II

32 篇文章 0 订阅
30 篇文章 0 订阅

原题链接在这里:https://leetcode.com/problems/unique-binary-search-trees-ii/

这道题是Unique Binary Search Trees的进阶版本. 返回的不是个数,而是每一个结果。

循环中递归调用helper函数求解子问题,以i为root时,BST的left child 是由1到i-1生成的,BST的right child 是由i+1 到n生成的。

然后从左右子树的返回结果中依次取点,接到有 i 生成的根上,一共有m*n种接法。构造好后当前root加到res里,走完大loop返回res.

当数组为 1,2,3,4,.. i,.. n时,基于以下原则的BST建树具有唯一性:
以i为根节点的树,其左子树由[1, i-1]构成, 其右子树由[i+1, n]构成。 

可行的BST数量是卡特兰数,不是多项式时间,所以要求解所有结果也不是在多项式时间内可以完成的。

AC Java:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<TreeNode> generateTrees(int n) {
        return helper(1,n);<span style="white-space:pre">	</span>//1 is set as begging, n is set as end.
    }
    private List<TreeNode> helper(int left, int right){
        List<TreeNode> res = new ArrayList<TreeNode>();
        if(left>right){
            res.add(null);
            return res;
        }
        for(int i = left; i <= right; i++){
            List<TreeNode> leftRes = helper(left,i-1);
            List<TreeNode> rightRes = helper(i+1,right);
            for(int m = 0; m < leftRes.size(); m++){
                for(int n = 0; n < rightRes.size(); n++){
                    TreeNode root = new TreeNode(i);
                    root.left = leftRes.get(m);
                    root.right = rightRes.get(n);
                    res.add(root);
                }
            }
        }
        return res;
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值