题目链接:57. 爬楼梯
题目描述
题目描述
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬至多m (1 <= m < n)个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
输入描述
输入共一行,包含两个正整数,分别表示n, m
输出描述
输出一个整数,表示爬到楼顶的方法数。
输入示例
3 2
输出示例
3
提示信息
数据范围:
1 <= m < n <= 32;当 m = 2,n = 3 时,n = 3 这表示一共有三个台阶,m = 2 代表你每次可以爬一个台阶或者两个台阶。
此时你有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶段
- 1 阶 + 2 阶
- 2 阶 + 1 阶
文章讲解:代码随想录
题解1:动态规划
思路:这是一个完全背包问题,物品为1到m,背包容量为 n。
动态规划分析:
- dp 数组以及下标的含义:dp[j] 代表爬到 j 阶楼梯有几种方法。
- 递推公式:dp[j] += dp[j - i]。
- dp 数组初始化:dp[0] = 1,其他全为0。
- 遍历顺序:本题目标是求排列数,因此先遍历背包,再遍历物品。
- 打印 dp 数组:
const readline = require('readline');
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout
});
rl.on("line", (row) => {
const vars = row.split(' ');
const n = parseInt(vars[0]);
const m = parseInt(vars[1]);
const dp = new Array(n + 1).fill(0);
dp[0] = 1;
for (let j = 1; j <= n; j++) {
for (let i = 1; i <= m; i++) {
if (j >= i) {
dp[j] += dp[j - i];
}
}
}
console.log(dp[n]);
});
分析:时间复杂度为 O(n * m),空间复杂度为 O(m)。
收获
用完全背包的思路求解爬楼梯。