从一道动态规划题带你领略『卡特兰数』是如何秒杀算法题的

苦逼的码农

共 885字,需浏览 2分钟

 ·

2019-12-08 23:31

来源公众号:苦逼的码农

作者:stul

一、 leetcode 96 号题

题目链接:https://leetcode-cn.com/problems/unique-binary-search-trees/

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

n = 3 时:

df511c32e2ad1be8fdc6a976fd42cb60.webp

二、动态规划

思路:从 1 开始到 n ,每次以这个数为根,左子树存放比它小的数,右子树存放比它大的数。每个根不重复,因此每个树也必定不重复。

左子树和右子树,又可以按照这个规则去生成新的树。

df511c32e2ad1be8fdc6a976fd42cb60.webp

例如:n = 3的时候

1为根:比 1 小的数只有 0,不用管。比 1 大的数有 2 和 3。

拿 2 和 3 来生成一棵树和拿 1 和 2 来生成一棵树的种数是不是相同的?那么 1 和 2 能生成多少种树呢?

2为根:比 2 小的是 1,比 2 大的是 3。这里只有 1 种。

3为根:比 3 小的是 1 和 2,1 和 2 能生成多少种树呢?

我们先暂停思维,来到一个新的问题。n = 2 的时候,结果应该是多少?

33a3961432feb9f74e554a566d114661.webp

n = 2 的时候,按照我们之前的方法。

1 为根:比 1 大的数只有 2, 这里有 1 种。
2 为根:比 2 小的数只有 1, 这里有 1 种。

那答案就应该是 2 种。

解决了 n = 2 的问题,那 n = 3 的问题就也解决了。ans = 2 + 1 + 2 = 5。

我们来看一般情况。输入一个 n 。

定义一个 F(i) 表示以 i 为根,生成的树的种数。

定义一个 G(n) 表示输入 n 的时候,输出的结果。此处一定要注意 F 与 G 的区别。

以 i 为根的时候,能生成多少种树?

比 i 小的有 i - 1 个,比 i 大的有 n - i 个。因此左子树有 i - 1 个, 右子树有 n - i 个数。那么,F(i) = G(i - 1) * G(n - i)。

而我们求的 G(n) = F(1) + F(2) + …… + F(n)。

把两个公式综合 :

96ddd722531d5161b51203b7aa43580e.webp


d[0] = 1// 0 的时候特殊处理
for (int i = 1; i <= n; i++)
    for (int j = 1; j <= i; j++) 
        d[i] += d[j-1] * d[i-j]; 

以上是利用动态规划求解的思路。

时间复杂度:O(n^2)

空间复杂度:O(n)

三、 Catalan公式

这个题目还有一种很强的解法,卡特兰公式。卡特兰公式和排列组合有很大关系,不属于偏难怪解法,有很多算法和数据结构的问题本质上就是卡特兰公式的应用。比如二叉树的形态数,出栈序列数,括号匹配问题等。公式不要紧,没必要去硬背。主要是理解卡特兰问题应用的特征,把问题抽象到已有模型中来。

Catalan 递推项满足:

C(n) = C(0) * C(n-1) + C(1) * C(n-2) + … + C(n-2) * C(1) + C(n-1) * C(0)

Catalan 通项公式:7b814b70e291b27a9c141b80b2a606d4.webp = c5b13ec998f0d110aec6f0cb89597e0a.webp 53bc4a7058720eb07e33c5a0c866d3de.webp

Catalan 递推公式1:445b93929d4e7e1dd4c24ea277eced69.webp = 3f87ce71f5331d35108c86717f367588.webp 7b814b70e291b27a9c141b80b2a606d4.webp

Catalan 性质:7b814b70e291b27a9c141b80b2a606d4.webp = 53bc4a7058720eb07e33c5a0c866d3de.webp -326fe13ac5a18d96778052ce4d4b575a.webp

这个题目里面,由我们上面的 G(n) 很容易可以看出是一个卡特兰的应用。

我们用它的递推公式来求解。

7b814b70e291b27a9c141b80b2a606d4.webp 的值,445b93929d4e7e1dd4c24ea277eced69.webp = 3f87ce71f5331d35108c86717f367588.webp 7b814b70e291b27a9c141b80b2a606d4.webp。公式顺推,代码中 n 次循环结束后的 ans 值就是 4ce2a9973f8bfd21449a7a2434f10d69.webp  对应的值。类比斐波那契最简解法。

    long ans = 1;
    for(int i = 1; i <= n; i++)
        ans = ans * 2 * (2 * i + 1) / (i + 2);
    return (int) ans;

时间复杂度:O(n)

空间复杂度:O(1)

四、Catalan应用

例题1:括号序列

给 n 对括号,可以合成的合法序列有多少种?

44fb2a922e112cb128b1148c599ec9b9.webp
  1. 首先计算一共的序列种数。n 对括号, 一共有 2n 个位置。我们选出其中 n 个位置放放置左括号,剩下的位置肯定就是右括号了。因此一共的种数为:53bc4a7058720eb07e33c5a0c866d3de.webp

  2. 接下来找出非法的括号序列数。每个非法的序列,在它的奇数位置,一定存在右括号数量大于左括号的数量。

b9aefe28a1c88299b6a826e5e9b9c430.webp

在上图中:第 2m + 1 的时候,右括号大于左括号,因此该序列非法。

在 2m + 1 前:

右括号数量为:m + 1

左括号数量为:m

在 2m + 1后:

总的数量:n - 2m - 1

左括号有:n - 2m - 1 - (m + 1) = n - m

右括号有:n - 2m - 1 - m = n - m - 1

这个时候我们将右边的左括号和右括号位置置换(总的组合数量不会受到影响)。那么在整个序列 2n 个位置中:

右括号的数量为:m + 1 + n -  m = n + 1

左括号的数量为:m + n - m - 1 = n - 1

在长度为 2n 里面有 n + 1 个右括号,数量为:3f4345f311a2d4d4ae71ced1ca20d2e2.webp 。你也可以理解从左括号的角度去看:326fe13ac5a18d96778052ce4d4b575a.webp

在上面两个步骤以后,我们得到的合法序列数:53bc4a7058720eb07e33c5a0c866d3de.webp  - 326fe13ac5a18d96778052ce4d4b575a.webp 。这就是 Catalan 公式的性质公式。知道是 Catalan,我们就可以用刚刚的方法求解问题的答案。

例题2 :一个栈(无穷大)的进栈序列为1,2,3,…,n,有多少个不同的出栈序列?

例题3 :给出一个n,要求一个长度为2n的01序列,使得序列的任意前缀中1的个数不少于0的个数, 有多少个不同的01序列?

例题4 :2n个人要买票价为五元的电影票,每人只买一张,但是售票员没有钱找零。其中,n个人持有五元,另外n个人持有十元,问在不发生找零困难的情况下,有多少种排队方法?(阿里有个笔试题根据这个变化的)

其他动态规划算法题推荐:

1、告别动态规划,连刷40道动规算法题,我总结了动规的套路

2、动态规划该如何优化?我总结了这些套路,以后优化就是分分钟

3、算法专题(动规):不同的定义产生不同的解法

4、详解三道一维的动态规划算法题

5、经典动态规划:高楼扔鸡蛋

6、详解 leetcode 221 题:最大正方形

7、动态规划之正则表达式

推荐阅读

写了很久,这是一份适合普通大众/科班/非科班的『学习路线』

有必要说一说即将到来的春招(经历+重要性+如何准备)

普普通通,我的三年大学

历经两个月,我的秋招之路结束了!


浏览 73
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报