卡特兰数之不同的二叉搜索树

在前面写斐波那契数列的时候,突然想起来卡特兰数列,所以特地来重温一下。

什么是卡特兰数列

首先这里给出它的递推公式  C= C0Cn-1 + C1Cn-2 + ... + Cn-1C0

咋一看,看不明白,这就对了,咱来举几个例子:

这里我们先把第一项直接写出来 C0 = 1,那么

C1C0C0 = 1 * 1 = 1

C2C0C1C1C0 = 1  + 1  = 2

C3 = C0C2 + C1C1 C2C0 = 2  + 1 + 2 = 5

...

高中数学到此为止,再写得露馅,回到今天的例题。

不同的二叉搜索树

问题:给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?

例:输入n = 3,则结果等于5。5种二叉搜索树如下:

分析:

这里我们注意一下二叉搜素树的性质,满足当前节点值大于左节点值,小于右节点值的二叉树才是二叉搜索树。

假设n个节点的二叉树搜索树的个数是Gn,以 i(1 <= i <= n) 为根节点的二叉搜索树个数定义为Fi

Gn FFF+ ... + Fn

根据搜索树性质,则当根节点的值为i时, 它的左子树为由 节点1 到 节点i-1 组成的二叉搜索树,右子树为由 节点i + 1 到 节点n 组成的二叉搜素树。

推到出: F= Gi-1 Gn-i

那么 GG0Gn-1 G1Gn-2 G2Gn-3 + ... + Gn-1G0

原来G就是卡塔兰数!

实现:

这里用的是DP,先计算G,再计算G,一直到Gn。 

 1 int CatalanNum(int n) {
 2     if(n <= 1){
 3         return 1;
 4     }
 5     int[] dp = new int[n + 1];
 6     dp[0] = 1;
 7     dp[1] = 1;
 8     for(int i = 2; i <= n; i++){
 9         for(int j = 1; j <= i; j++){
10             dp[i] += dp[j - 1] * dp[i - j];
11         }
12     }
13     return dp[n];
14 }

时间复杂度:(2 + n) * (n - 1) / 2, 也就是O(N2)。

空间复杂度:dp数组,O(N)。

通项公式:

果然,数学家又给出了通项公式:

 代码实现:

int CatalanNum (int n) {
    if(n == 0){
        return 1;
    }
    int[] dp = new int[n + 1];
    dp[0] = 1;
    for(int i = 0; i < n; i++){
        dp[i + 1] = dp[i] * 2*(2 * i + 1) / (i + 2);
    }
    return dp[n];
}

时间复杂度:(O(N)。

空间复杂度:dp数组,O(N)。(可以优化到O(1), 为什么我每次都不优化。。。)

最后提醒一下:卡特兰数增长非常快,建议使用大数类型。

PS:

最近看到很多简历都提到了Dijkstra,哇塞,真的很亮眼。下次写它。

End

猜你喜欢

转载自www.cnblogs.com/benzhai/p/12423769.html