给出一个 n, 用二叉搜索树来存储1 … n,总共有多少种不同构的二叉搜索树?
Example:
输入: 3
输出: 5
解释:
给出 n = 3, 总共有5种不同构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
开始想的是搜索,时间复杂度太高。然后查了一下,这是个卡特兰数的应用场景。
什么是卡特兰数?
卡特兰数是满足h(n)=h(0)h(n-1)+h(1)h(n-2)+...+h(n-1)h(0)
递推公式的数。
这个问题为什么是卡特兰数,因为它满足上述的递推公式。下面给出简单的推导过程。
首先,我们假设n个数1 … n,能够形成的不同的二叉搜索树的个数是f(n)。
考虑,
· 当n为1时,f(1)=?,因为只有一个节点,显然f(1)=1;
· 当n为2时,有2个节点。
考虑根节点可以为1,那么形成的二叉树为
1
\
2
根节点也可以为2,那么形成的二叉树为
2
/
1
>>因此f(2)=2;
·当n为3时,有3个节点。
1.考虑根节点可以为1,那么形成的二叉树为
1 1
\ \
2 或者 3
\ /
3 2
2.根节点也可以为2,那么形成的二叉树为
2
/ \
1 3
3.考虑根节点可以为3,那么形成的二叉树为
3 3
/ /
2 1
/ \
1 2
>>因此f(3)=5;
考虑普遍情况,对于n个节点,根节点为k时:
根节点的左子树是一颗独立的树,同样右子树也是一颗独立的树。
那么当根节点k=1时,它的左子树有0个节点,右子树有n-1个节点(根节点已经占了一个节点)。
·所以对于该左子树,就是计算0个节点的不同的二叉搜索树个数的问题,也就是f(0)。
·对于右子树,就是计算n-1个节点的不同的二叉搜索树个数的问题,也就是f(n-1)。
>>由于左右子树是独立的问题,所以当根节点k=1时,总个数为左子树的不同BST个数*右子树不同BST个数,
也就是f(0)*f(n-1)
上述只计算了根节点k=1的情况,现在根节点可以取[1,n]之间的任意一个整数。
因此,最后的结果也就是fh(n)=f(0)f(n-1)+f(1)f(n-2)+...+f(n-1)f(0)
这与卡特兰数的递推公式相同,因此这是个计算卡特兰数的问题。
那么问题来了,上边我们只分析了这个问题属于卡特兰数问题,如何计算卡特兰数呢?
由于我们已经有了递推公式,只需写一个程序来计算即可:
public static int numTrees(int n) {
if(n<0) return 0;
int [] f=new int[n+1];
f[0]=1;
for(int i=1;i<=n;i++){
for (int j = 0; j <i ; j++) {
f[i]+=f[j]*f[i-1-j];
}
}
return f[n];
}
或者也可以通过通项公式来进行计算,不过我个人暂时觉得递推的方法好些。
下面也给出通项公式(转载一下吧):
通项公式
还有一些扩展的问题也是属于卡特兰数的,比如一个栈(无穷大)的进栈序列为1,2,3,…n,有多少个不同的出栈序列?待补充。。。