假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
- 按照动态规划五步法,第一步应该确定dp数组每个元素的含义,dp[i]代表到第i阶楼梯有多少种方法。
- 第二步应该确定递推公式,由于每次只能爬一阶或者两阶,所以dp[i]可以由dp[i - 1]和的dp[i - 2]得到,因为dp[i - 1]是到第i - 1阶有多少种方法,再爬一阶就是第i阶了,dp[i - 2]是到第i - 2阶有多少种方法,再一次性爬两阶就是第i阶了。所以dp[i] = dp[i - 1] + dp[i - 2]。
- 这里n不会等于0,所以dp[0]并没有什么意义,很容易得到dp[1]=1,dp[2] = 2。
- 然后遍历从第三阶开始,就有以下代码。
int climbStairs(int n){
if (n == 1)
return 1;
int* dp = malloc(sizeof(int) * (n + 1));
dp[1] = 1;
dp[2] = 2;
for (int i = 3; i <= n; i++) {
dp[i] = dp[i - 1] + dp[i - 2];
}
return dp[n];
}
优化一下空间复杂度:
int climbStairs(int n){
if (n == 1)
return 1;
int dp[3];
dp[1] = 1;
dp[2] = 2;
for (int i = 3; i <= n; i++) {
int sum = dp[1] + dp[2];
dp[1] = dp[2];
dp[2] = sum;
}
return dp[2];
}
- 这种题目的进阶版是一次可以爬m个台阶,这种属于完全背包问题,后续再进行补充。