( Leetcode 95 ) Unique Binary Search Trees II

题目:Unique Binary Search Trees II

Given an integer 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
解题思路:

首先这道题和 Unique Binary Search Trees 有一部分相似,都是求给定结点n BST树能够构造多少棵不同的树。第一道题运用的是动态规划,我参考的这篇博客:http://www.cnblogs.com/springfor/p/3884009.html,具体的看这篇博客的讲解。下面重点说说本道题的思路。

首先使用动态规划计算节点从1 到 n 可以构造的 BST 树的个数,接下来使用递归就可以构造一棵 BST 树了,基本思路是 以 i 为根节点,然后 [1, i] 区间构造左子树,[i+1,n] 区间构造右子树,如此不断递归就构造成了一棵完整二叉树。

具体看下面的 Java 代码:

public class Solution {
	// s = start ,e = end
    public TreeNode[] recursion (int s, int e, int[] dp) {
        TreeNode[] roots = null;
        int curlen = 0;
        if (s > e) {
            roots = new TreeNode[1];
            roots[0] = null;
            return roots;
        }
        roots = new TreeNode[dp[e - s + 1]];
        for (int i = s; i <= e; i++) {
            TreeNode[] lefts = recursion(s, i - 1, dp);  //递归构造左子树
            TreeNode[] rights = recursion(i + 1, e, dp);  //递归构造右子树
            for (TreeNode left : lefts) {
                for (TreeNode right : rights) {
                    TreeNode root = new TreeNode(i);
                    root.left = left;
                    root.right = right;
                    roots[curlen++] = root;
                }
            }
        }
        return roots;
    }
	
    public List<TreeNode> generateTrees(int n) {
        if (n == 0) {
			return new ArrayList<TreeNode>();
		} else {
            int[] dp = new int[n + 1];
            dp[0] = 1;
            dp[1] = 1;
            for (int i = 2; i <= n; i++) {
				for (int j = 1; j <= i; j++) {
					dp[i] += dp[j - 1] * dp[i - j];
				}   
			}   
            TreeNode[] resarr =  recursion(1, n, dp);
            List<TreeNode> res = new ArrayList<>();
            for (TreeNode node : resarr) {
                res.add(node);
            }
            return res;
        }
    }
}

程序运行时间 2 ms。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值