mirror of
https://github.com/krahets/hello-algo.git
synced 2026-04-13 18:00:18 +08:00
build
This commit is contained in:
@@ -10,7 +10,7 @@ comments: true
|
||||
|
||||
!!! question "爬楼梯"
|
||||
|
||||
给定一个共有 $n$ 阶的楼梯,你每步可以上 $1$ 阶或者 $2$ 阶,请问有多少种方案可以爬到楼顶。
|
||||
给定一个共有 $n$ 阶的楼梯,你每步可以上 $1$ 阶或者 $2$ 阶,请问有多少种方案可以爬到楼顶?
|
||||
|
||||
如图 14-1 所示,对于一个 $3$ 阶楼梯,共有 $3$ 种方案可以爬到楼顶。
|
||||
|
||||
@@ -18,7 +18,7 @@ comments: true
|
||||
|
||||
<p align="center"> 图 14-1 爬到第 3 阶的方案数量 </p>
|
||||
|
||||
本题的目标是求解方案数量,**我们可以考虑通过回溯来穷举所有可能性**。具体来说,将爬楼梯想象为一个多轮选择的过程:从地面出发,每轮选择上 $1$ 阶或 $2$ 阶,每当到达楼梯顶部时就将方案数量加 $1$ ,当越过楼梯顶部时就将其剪枝。
|
||||
本题的目标是求解方案数量,**我们可以考虑通过回溯来穷举所有可能性**。具体来说,将爬楼梯想象为一个多轮选择的过程:从地面出发,每轮选择上 $1$ 阶或 $2$ 阶,每当到达楼梯顶部时就将方案数量加 $1$ ,当越过楼梯顶部时就将其剪枝。代码如下所示:
|
||||
|
||||
=== "Python"
|
||||
|
||||
@@ -39,7 +39,7 @@ comments: true
|
||||
|
||||
def climbing_stairs_backtrack(n: int) -> int:
|
||||
"""爬楼梯:回溯"""
|
||||
choices = [1, 2] # 可选择向上爬 1 或 2 阶
|
||||
choices = [1, 2] # 可选择向上爬 1 阶或 2 阶
|
||||
state = 0 # 从第 0 阶开始爬
|
||||
res = [0] # 使用 res[0] 记录方案数量
|
||||
backtrack(choices, state, n, res)
|
||||
@@ -67,7 +67,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
int climbingStairsBacktrack(int n) {
|
||||
vector<int> choices = {1, 2}; // 可选择向上爬 1 或 2 阶
|
||||
vector<int> choices = {1, 2}; // 可选择向上爬 1 阶或 2 阶
|
||||
int state = 0; // 从第 0 阶开始爬
|
||||
vector<int> res = {0}; // 使用 res[0] 记录方案数量
|
||||
backtrack(choices, state, n, res);
|
||||
@@ -96,7 +96,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
int climbingStairsBacktrack(int n) {
|
||||
List<Integer> choices = Arrays.asList(1, 2); // 可选择向上爬 1 或 2 阶
|
||||
List<Integer> choices = Arrays.asList(1, 2); // 可选择向上爬 1 阶或 2 阶
|
||||
int state = 0; // 从第 0 阶开始爬
|
||||
List<Integer> res = new ArrayList<>();
|
||||
res.add(0); // 使用 res[0] 记录方案数量
|
||||
@@ -126,7 +126,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
int ClimbingStairsBacktrack(int n) {
|
||||
List<int> choices = [1, 2]; // 可选择向上爬 1 或 2 阶
|
||||
List<int> choices = [1, 2]; // 可选择向上爬 1 阶或 2 阶
|
||||
int state = 0; // 从第 0 阶开始爬
|
||||
List<int> res = [0]; // 使用 res[0] 记录方案数量
|
||||
Backtrack(choices, state, n, res);
|
||||
@@ -157,7 +157,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
func climbingStairsBacktrack(n int) int {
|
||||
// 可选择向上爬 1 或 2 阶
|
||||
// 可选择向上爬 1 阶或 2 阶
|
||||
choices := []int{1, 2}
|
||||
// 从第 0 阶开始爬
|
||||
state := 0
|
||||
@@ -190,7 +190,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
func climbingStairsBacktrack(n: Int) -> Int {
|
||||
let choices = [1, 2] // 可选择向上爬 1 或 2 阶
|
||||
let choices = [1, 2] // 可选择向上爬 1 阶或 2 阶
|
||||
let state = 0 // 从第 0 阶开始爬
|
||||
var res: [Int] = []
|
||||
res.append(0) // 使用 res[0] 记录方案数量
|
||||
@@ -218,7 +218,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
function climbingStairsBacktrack(n) {
|
||||
const choices = [1, 2]; // 可选择向上爬 1 或 2 阶
|
||||
const choices = [1, 2]; // 可选择向上爬 1 阶或 2 阶
|
||||
const state = 0; // 从第 0 阶开始爬
|
||||
const res = new Map();
|
||||
res.set(0, 0); // 使用 res[0] 记录方案数量
|
||||
@@ -251,7 +251,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
function climbingStairsBacktrack(n: number): number {
|
||||
const choices = [1, 2]; // 可选择向上爬 1 或 2 阶
|
||||
const choices = [1, 2]; // 可选择向上爬 1 阶或 2 阶
|
||||
const state = 0; // 从第 0 阶开始爬
|
||||
const res = new Map();
|
||||
res.set(0, 0); // 使用 res[0] 记录方案数量
|
||||
@@ -281,7 +281,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
int climbingStairsBacktrack(int n) {
|
||||
List<int> choices = [1, 2]; // 可选择向上爬 1 或 2 阶
|
||||
List<int> choices = [1, 2]; // 可选择向上爬 1 阶或 2 阶
|
||||
int state = 0; // 从第 0 阶开始爬
|
||||
List<int> res = [];
|
||||
res.add(0); // 使用 res[0] 记录方案数量
|
||||
@@ -309,7 +309,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
fn climbing_stairs_backtrack(n: usize) -> i32 {
|
||||
let choices = vec![ 1, 2 ]; // 可选择向上爬 1 或 2 阶
|
||||
let choices = vec![ 1, 2 ]; // 可选择向上爬 1 阶或 2 阶
|
||||
let state = 0; // 从第 0 阶开始爬
|
||||
let mut res = Vec::new();
|
||||
res.push(0); // 使用 res[0] 记录方案数量
|
||||
@@ -340,7 +340,7 @@ comments: true
|
||||
|
||||
/* 爬楼梯:回溯 */
|
||||
int climbingStairsBacktrack(int n) {
|
||||
int choices[2] = {1, 2}; // 可选择向上爬 1 或 2 阶
|
||||
int choices[2] = {1, 2}; // 可选择向上爬 1 阶或 2 阶
|
||||
int state = 0; // 从第 0 阶开始爬
|
||||
int *res = (int *)malloc(sizeof(int));
|
||||
*res = 0; // 使用 res[0] 记录方案数量
|
||||
@@ -375,7 +375,7 @@ comments: true
|
||||
|
||||
// 爬楼梯:回溯
|
||||
fn climbingStairsBacktrack(n: usize) !i32 {
|
||||
var choices = [_]i32{ 1, 2 }; // 可选择向上爬 1 或 2 阶
|
||||
var choices = [_]i32{ 1, 2 }; // 可选择向上爬 1 阶或 2 阶
|
||||
var state: i32 = 0; // 从第 0 阶开始爬
|
||||
var res = std.ArrayList(i32).init(std.heap.page_allocator);
|
||||
defer res.deinit();
|
||||
@@ -387,15 +387,15 @@ comments: true
|
||||
|
||||
## 14.1.1 方法一:暴力搜索
|
||||
|
||||
回溯算法通常并不显式地对问题进行拆解,而是将问题看作一系列决策步骤,通过试探和剪枝,搜索所有可能的解。
|
||||
回溯算法通常并不显式地对问题进行拆解,而是将求解问题看作一系列决策步骤,通过试探和剪枝,搜索所有可能的解。
|
||||
|
||||
我们可以尝试从问题分解的角度分析这道题。设爬到第 $i$ 阶共有 $dp[i]$ 种方案,那么 $dp[i]$ 就是原问题,其子问题包括:
|
||||
我们可以尝试从问题分解的角度分析这道题。设爬到第 $i$ 阶共有 $dp[i]$ 种方案,那么 $dp[i]$ 就是原问题,其子问题包括:
|
||||
|
||||
$$
|
||||
dp[i-1], dp[i-2], \dots, dp[2], dp[1]
|
||||
$$
|
||||
|
||||
由于每轮只能上 $1$ 阶或 $2$ 阶,因此当我们站在第 $i$ 阶楼梯上时,上一轮只可能站在第 $i - 1$ 阶或第 $i - 2$ 阶上。换句话说,我们只能从第 $i -1$ 阶或第 $i - 2$ 阶前往第 $i$ 阶。
|
||||
由于每轮只能上 $1$ 阶或 $2$ 阶,因此当我们站在第 $i$ 阶楼梯上时,上一轮只可能站在第 $i - 1$ 阶或第 $i - 2$ 阶上。换句话说,我们只能从第 $i -1$ 阶或第 $i - 2$ 阶迈向第 $i$ 阶。
|
||||
|
||||
由此便可得出一个重要推论:**爬到第 $i - 1$ 阶的方案数加上爬到第 $i - 2$ 阶的方案数就等于爬到第 $i$ 阶的方案数**。公式如下:
|
||||
|
||||
@@ -411,7 +411,7 @@ $$
|
||||
|
||||
我们可以根据递推公式得到暴力搜索解法。以 $dp[n]$ 为起始点,**递归地将一个较大问题拆解为两个较小问题的和**,直至到达最小子问题 $dp[1]$ 和 $dp[2]$ 时返回。其中,最小子问题的解是已知的,即 $dp[1] = 1$、$dp[2] = 2$ ,表示爬到第 $1$、$2$ 阶分别有 $1$、$2$ 种方案。
|
||||
|
||||
观察以下代码,它和标准回溯代码都属于深度优先搜索,但更加简洁。
|
||||
观察以下代码,它和标准回溯代码都属于深度优先搜索,但更加简洁:
|
||||
|
||||
=== "Python"
|
||||
|
||||
@@ -644,7 +644,7 @@ $$
|
||||
|
||||
<p align="center"> 图 14-3 爬楼梯对应递归树 </p>
|
||||
|
||||
观察图 14-3 ,**指数阶的时间复杂度是由于“重叠子问题”导致的**。例如 $dp[9]$ 被分解为 $dp[8]$ 和 $dp[7]$ ,$dp[8]$ 被分解为 $dp[7]$ 和 $dp[6]$ ,两者都包含子问题 $dp[7]$ 。
|
||||
观察图 14-3 ,**指数阶的时间复杂度是“重叠子问题”导致的**。例如 $dp[9]$ 被分解为 $dp[8]$ 和 $dp[7]$ ,$dp[8]$ 被分解为 $dp[7]$ 和 $dp[6]$ ,两者都包含子问题 $dp[7]$ 。
|
||||
|
||||
以此类推,子问题中包含更小的重叠子问题,子子孙孙无穷尽也。绝大部分计算资源都浪费在这些重叠的问题上。
|
||||
|
||||
@@ -655,6 +655,8 @@ $$
|
||||
1. 当首次计算 $dp[i]$ 时,我们将其记录至 `mem[i]` ,以便之后使用。
|
||||
2. 当再次需要计算 $dp[i]$ 时,我们便可直接从 `mem[i]` 中获取结果,从而避免重复计算该子问题。
|
||||
|
||||
代码如下所示:
|
||||
|
||||
=== "Python"
|
||||
|
||||
```python title="climbing_stairs_dfs_mem.py"
|
||||
@@ -973,7 +975,7 @@ $$
|
||||
}
|
||||
```
|
||||
|
||||
观察图 14-4 ,**经过记忆化处理后,所有重叠子问题都只需被计算一次,时间复杂度被优化至 $O(n)$** ,这是一个巨大的飞跃。
|
||||
观察图 14-4 ,**经过记忆化处理后,所有重叠子问题都只需计算一次,时间复杂度优化至 $O(n)$** ,这是一个巨大的飞跃。
|
||||
|
||||
{ class="animation-figure" }
|
||||
|
||||
@@ -981,11 +983,11 @@ $$
|
||||
|
||||
## 14.1.3 方法三:动态规划
|
||||
|
||||
**记忆化搜索是一种“从顶至底”的方法**:我们从原问题(根节点)开始,递归地将较大子问题分解为较小子问题,直至解已知的最小子问题(叶节点)。之后,通过回溯将子问题的解逐层收集,构建出原问题的解。
|
||||
**记忆化搜索是一种“从顶至底”的方法**:我们从原问题(根节点)开始,递归地将较大子问题分解为较小子问题,直至解已知的最小子问题(叶节点)。之后,通过回溯逐层收集子问题的解,构建出原问题的解。
|
||||
|
||||
与之相反,**动态规划是一种“从底至顶”的方法**:从最小子问题的解开始,迭代地构建更大子问题的解,直至得到原问题的解。
|
||||
|
||||
由于动态规划不包含回溯过程,因此只需使用循环迭代实现,无须使用递归。在以下代码中,我们初始化一个数组 `dp` 来存储子问题的解,它起到了记忆化搜索中数组 `mem` 相同的记录作用。
|
||||
由于动态规划不包含回溯过程,因此只需使用循环迭代实现,无须使用递归。在以下代码中,我们初始化一个数组 `dp` 来存储子问题的解,它起到了与记忆化搜索中数组 `mem` 相同的记录作用:
|
||||
|
||||
=== "Python"
|
||||
|
||||
@@ -1233,17 +1235,17 @@ $$
|
||||
|
||||
<p align="center"> 图 14-5 爬楼梯的动态规划过程 </p>
|
||||
|
||||
与回溯算法一样,动态规划也使用“状态”概念来表示问题求解的某个特定阶段,每个状态都对应一个子问题以及相应的局部最优解。例如,爬楼梯问题的状态定义为当前所在楼梯阶数 $i$ 。
|
||||
与回溯算法一样,动态规划也使用“状态”概念来表示问题求解的特定阶段,每个状态都对应一个子问题以及相应的局部最优解。例如,爬楼梯问题的状态定义为当前所在楼梯阶数 $i$ 。
|
||||
|
||||
根据以上内容,我们可以总结出动态规划的常用术语。
|
||||
|
||||
- 将数组 `dp` 称为「$dp$ 表」,$dp[i]$ 表示状态 $i$ 对应子问题的解。
|
||||
- 将最小子问题对应的状态(即第 $1$ 和 $2$ 阶楼梯)称为「初始状态」。
|
||||
- 将最小子问题对应的状态(第 $1$ 阶和第 $2$ 阶楼梯)称为「初始状态」。
|
||||
- 将递推公式 $dp[i] = dp[i-1] + dp[i-2]$ 称为「状态转移方程」。
|
||||
|
||||
## 14.1.4 空间优化
|
||||
|
||||
细心的你可能发现,**由于 $dp[i]$ 只与 $dp[i-1]$ 和 $dp[i-2]$ 有关,因此我们无须使用一个数组 `dp` 来存储所有子问题的解**,而只需两个变量滚动前进即可。
|
||||
细心的读者可能发现了,**由于 $dp[i]$ 只与 $dp[i-1]$ 和 $dp[i-2]$ 有关,因此我们无须使用一个数组 `dp` 来存储所有子问题的解**,而只需两个变量滚动前进即可。代码如下所示:
|
||||
|
||||
=== "Python"
|
||||
|
||||
@@ -1445,6 +1447,6 @@ $$
|
||||
}
|
||||
```
|
||||
|
||||
观察以上代码,由于省去了数组 `dp` 占用的空间,因此空间复杂度从 $O(n)$ 降低至 $O(1)$ 。
|
||||
观察以上代码,由于省去了数组 `dp` 占用的空间,因此空间复杂度从 $O(n)$ 降至 $O(1)$ 。
|
||||
|
||||
在动态规划问题中,当前状态往往仅与前面有限个状态有关,这时我们可以只保留必要的状态,通过“降维”来节省内存空间。**这种空间优化技巧被称为“滚动变量”或“滚动数组”**。
|
||||
|
||||
Reference in New Issue
Block a user