Leetcode(动态规划):爬楼梯
题目描述
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?1
判题标准
系统会用下面的代码来测试你的题解:
示例 1:
输入:n = 2
输出:2
解释:有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
示例 2:
输入:n = 3
输出:3
解释:有三种方法可以爬到楼顶。
4. 1 阶 + 1 阶 + 1 阶
5. 1 阶 + 2 阶
6. 2 阶 + 1 阶
解题思路——动态规划(C++)
当爬n个台阶到楼顶时,所有方法的数量为爬到n-1个台阶的方法数加上爬到n-2个台阶的方法数,则:
动态方程:dp[n] = dp[n-1]+dp[n-2]
若令dp[0]=1,满足dp[2]=2,则可将初始条件dp[1]=1、dp[2]=2简化为dp[0]=dp[1]=1
以上的思路由结果倒推,此外也可以由条件正推,较为容易理解,当爬到楼顶一共有3个台阶时,不难发现,可以由爬到1个台阶时再爬2个台阶,或者由爬到2个台阶时再爬1个台阶到达楼顶,那么爬到楼顶时的方法为爬1个台阶的方法数与爬2个台阶的方法数之和,此种方式可由滚动数组实现。
C++代码
class Solution {
public:
int climbStairs(int n) {
vector<int> dp(n+1);
dp[0]=dp[1]=1;
for(int i=2;i<=n;i++){
dp[i] = dp[i-1]+dp[i-2];
}
return dp[n];
}
};
class Solution {
public:
int climbStairs(int n) {
int a = 0,b=0,c=1;
for(int i=1;i<=n;i++){
a=b;
b=c;
c=a+b;
}
return c;
}
};