Given n, how many structurally unique BST's (binary search trees) that store values 1...n?
题意:给定整数n,计算出这1~n个整数能组成多少二叉查找树。
此题要做的事情跟上一题是一样的,只是要求的内容不同。大家可以先参考我上一题的思路 点击打开链接
利用上一题的方法中得到的List求size就可以得到个数,但是这样复杂度显然比较高。下面介绍我的另一种解法(其实思路是一样的,只是实现起来稍有不同)
首先,对于任意一颗二叉查找树,我们可以对它进行分类,分类的依据就是这棵树拥有多少个直接右节点。所谓的“直接右节点”就是指从根节点开始,递归寻找右节点,直至某个右节点无右节点。例如:一棵树只有左子树,那么它的直接右节点为0.
下面,我们分别对n=1,2,3,4,5时所能组成的二叉查找树进行分类:
n | 直接右节点 | 树的数量 | 总计 |
1 | 0 | 1 | 1 |
2 | 0 | 1 | |
1 | 1 | 2 | |
3 | 0 | 2 | |
1 | 2 | ||
2 | 1 | 5 | |
4 | 0 | 5 | |
1 | 5 | ||
2 | 3 | ||
3 | 1 | 14 | |
5 | 0 | 14 | |
1 | 14 | ||
2 | 9 | ||
3 | 4 | ||
4 | 1 | 42 |
1.对于n=k的情况,所有的二叉查找树中,最多只可能拥有直接右节点为k-1的树,且数量必定为1.(例如n=5时,直接右节点最大为4,且树的数量为1)
2.对于n=k的情况,所有的二叉查找树中,直接右节点为0的树的数量必定等于n=k-1时所拥有的树的总计数量。(例如n=5时,直接右节点为0的树的数量等于14,14是n=4时总计的树的数量)
3对于n=k的情况,所有的二叉查找树中,直接右节点数量为x(x属于区间(0,k-1))的树的数量等于 n=k-1时直接右节点数量为x-1,x……k-2的树的数量之和。(例如n=5时,直接右节点为2的树的数量=5+3+1)
现在,我们知道了从n-1到n时,二叉查找树数量的变换规律,下面,用代码实现它:
public class Solution {
public int numTrees(int n) {
int[] a = new int[n];
int[] b = new int[n];
if(n==0) return 0 ;
if(n==1) return 1;
a[0]=1;
for(int i=2;i<=n;i++){ //计算每个类别树的数量,存储在数组a中
b[0] = arrSum(a);
for(int j=1;j<n-1;j++){
b[j] = arrSum1(a,j);
}
b[i-1]=1;
int k = 0;
for(int m : b){
a[k++] = m;
}
}
return arrSum(a);
}
public int arrSum(int[] a){ //计算数组元素总和
int m = 0;
for(int i : a){
m+=i;
}
return m;
}
public int arrSum1(int[] a,int j){ //计算从下标j-1开始,到数组末尾的元素总和
int m = 0;
for(int i=j-1;i<a.length;i++){
m+=a[i];
}
return m;
}
}