动态规划——爬楼梯 Leetcode 70
解题思路
1、本问题可以分成多个子问题,爬第n阶楼梯的方法数量,等于两部分之和。即爬上n−1 阶楼梯的方法数量。因为再爬1阶就能到第n阶;爬上 n−2 阶楼梯的方法数量,因为再爬2阶就能到第n阶。
所以我们得到公式 dp[n] = dp[n-1] + dp[n-2] 。因为如果最后一步是1个台阶,那么求 n-1 台阶需要的步数就可以了,如果最后一步是2个台阶,就算n-2台阶需要的步数,因为只有2个可能,所以只需要算 dp[n-1]+dp[n-2],就是 dp[n] 。
2、同时需要初始化 dp[0]=1和 dp[1]=1 。
3、时间复杂度:O(n) 。
代码
class Solution {
public int climbStairs(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 1;
for(int i = 2; i <= n; i++) {
dp[i] = dp[i - 1] + dp[i - 2];
}
return dp[n];
}
}