首页 > 代码库 > Unique Binary Search Tree | LeetCode
Unique Binary Search Tree | LeetCode
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
通过仔细分析题目的规律,我们发现可以尝试通过动态规划的方法去解决。假设f(n)表示不同的二叉搜索树的个数,假设二叉搜索树的根为k(1<=k<=n),左子树的个数为k-1个,右子树的个数为n-k,左右子树分别都是二叉搜索树,因此依照定义,左子树的不同个数为f(k-1),右子树的个数为f(n-k),则根为k时的不同的二叉搜索树的个数为f(k-1)f(n-k),考虑到k的取值氛围,则所有的二叉搜索树的个数为f(n)=∑f(k-1)f(n-k)(k=1,2,...,n)。
得到递归的公式之后,我们可以很容易的写出程序:
Java:
public class Solution { public int numTrees(int n){ int f[] = new int[n+1]; f[0] = 1; for (int i = 1; i < f.length; i++) { for (int j = 1; j <= i; j++) { f[i] += f[j-1]*f[i-j]; } } return f[n]; }}
上面的程序是可以ac的,但是时间复杂度是O(n2),空间复杂度是O(n)。下面我们再介绍一种时间效率更高的方法,时间复杂度为O(n)。这里要引入一个叫Catalan(卡塔兰)数的概念,如果C(0)=1,C(1)=1,C(2)=2,...,C(n) = C(0)*C(n-1)+C(1)*C(n-2)+C(2)*C(n-2)+...+C(n-1)*C(0)=C(2n,n)/(n+1) (n=0,1,2,3....)
另类地推公式
C(n) = C(n-1)*(4n-2)/(n+1) (n=0,1,2,3....)
或者 C(n) = C(2n,n)-C(2n,n-1) (n=0,1,2,3...)
通过测试上面我们的几个例子发现,我们的所要求的二叉搜索树的个数就是卡塔兰数。
Java:
public class Solution{ public int numTrees(int n){ int answer = 1; for (int i = 1; i <= n; i++) { answer = (answer*(4*i-2))/(i+1); } return answer; }}
Unique Binary Search Tree | LeetCode
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。