题意描述:
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
动态规划,定义一个数组dp[],dp[i]: 爬到第i层楼梯,有dp[i]种方法。
首先是dp[i - 1],上i-1层楼梯,有dp[i - 1]种方法;还有就是dp[i - 2],上i-2层楼梯,有dp[i - 2]种方法。那么dp[i]就是 dp[i - 1]与dp[i - 2]之和,所以dp[i] = dp[i - 1] + dp[i - 2] 。
完整的C++代码如下:
class Solution {
public:
int climbStairs(int n) {
if(n <= 1) return n;
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];
}
};