• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
neverlandly
博客园    首页    新随笔    联系   管理    订阅  订阅

Leetcode: Unique Binary Search Trees

Given n, how many structurally unique BST's (binary search trees) that store values 1...n?

For example,
Given n = 3, there are a total of 5 unique BST's.

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

Analysis: 这道题是想的比较难,想通了的话,就好做了。我查看了网上的资料,发现别人的想法实在是太精秒了。http://blog.csdn.net/linhuanmars/article/details/24761459 以及 http://blog.csdn.net/u010500263/article/details/17748519

方法一:recursion

最基本是得先清楚BST定义. 每个节点的所有左子节点值都比该节点小, 每个节点的所有右子节点值都比该节点小. 

知道定义后在构建BST时可逐次选择value做根节点.  利用递归计算根节点左子树可能形状数目, 与根节点右子树可能形状数目, 最后将两个可能形状数目相乘得到所有可能的树形状.

 1 public class Solution {
 2     public int numTrees(int n) {
 3         int result = 0;
 4         if (n == 0 || n == 1) return 1;
 5         if (n < 0) return 0;
 6         for (int i = 0; i <= n - 1; i++) {
 7             result += numTrees(i) * numTrees(n-i-1);
 8         }
 9         return result;
10     }
11 }

方法二: DP

这道题要求可行的二叉查找树的数量,其实二叉查找树可以任意取根,只要满足中序遍历有序的要求就可以。从处理子问题的角度来看,选取一个结点为根,就把结点切成左右子树,以这个结点为根的可行二叉树数量就是左右子树可行二叉树数量的乘积,所以总的数量是将以所有结点为根的可行结果累加起来。写成表达式如下:

熟悉卡特兰数的朋友可能已经发现了,这正是卡特兰数的一种定义方式,是一个典型的动态规划的定义方式(根据其实条件和递推式求解结果)。所以思路也很明确了,维护量res[i]表示含有i个结点的二叉查找树的数量。根据上述递推式依次求出1到n的的结果即可。
时间上每次求解i个结点的二叉查找树数量的需要一个i步的循环,总体要求n次,所以总时间复杂度是O(1+2+...+n)=O(n^2)。空间上需要一个数组来维护,并且需要前i个的所有信息,所以是O(n)。代码如下:
 1 public int numTrees(int n) {
 2     if(n<=0)
 3         return 0;
 4     int[] res = new int[n+1];
 5     res[0] = 1;
 6     res[1] = 1;
 7     for(int i=2;i<=n;i++)
 8     {
 9         for(int j=0;j<i;j++)
10         {
11             res[i] += res[j]*res[i-j-1];
12         }
13     }
14     return res[n];
15 }

 Thinkagain: recursion的时间复杂度怎么计算?

posted @ 2014-06-26 12:26  neverlandly  阅读(234)  评论(0)    收藏  举报
刷新页面返回顶部
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3