一、题目描述
假设你正在爬楼梯。需要
n
阶你才能到达楼顶。每次你可以爬
1
或2
个台阶。你有多少种不同的方法可以爬到楼顶呢?示例 1:
输入:n = 2 输出:2 解释:有两种方法可以爬到楼顶。 1. 1 阶 + 1 阶 2. 2 阶示例 2:
输入:n = 3 输出:3 解释:有三种方法可以爬到楼顶。 1. 1 阶 + 1 阶 + 1 阶 2. 1 阶 + 2 阶 3. 2 阶 + 1 阶提示:
1 <= n <= 45
二、解题思路
典型的DP问题,先流程搞一波
楼顶有n个台阶,每次可以爬1个或2个,问题可以转化为 n-1个台阶的不同方法(对应下一步爬1个台阶),加上n-2个台阶的不同方法(对应下一步爬2个台阶),即F(n)=F(n-1)+F(n-2)
当只有0个台阶,那么只有0中方法,即F(0)=0
当只有1个台阶,那么只有1种方法,即F(1)=1
当只有2个台阶,那么只有2种方法,即F(2)=2
当只有3个台阶,那么只有F(3)=F(2)+F(1)=2+1=3
当有n个台阶,那么有F(n)=F(n-1+F(n-2) 种走法
2-1 状态定义
dp[i] 为 i 个台阶的走法
2-2 状态转移方程
dp[i]=dp[i-1]+dp[i-2]
2-3 遍历顺序
后依赖前,所以正序
2-4 初始化
正序遍历,且需要前2个值,初始化dp数组,至少需要两个值,dp[0]为特殊情况,单独处理,所以我们至少需要提前设置2个值。
dp[1]=1;
dp[2]=2;
2-5 输出结果
dp[n] 即为n个台阶的走法
2-6 优化
对于dp[i] 只会使用dp[i-1] 和 dp[i-2] 两个值,所以空间上可以优化成O(1)
三、Java代码
优化前
public int climbStairs(int n) {
if(n<=2) return n;
int[] dp=new 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];
}
优化后
class Solution {
public int climbStairs(int n) {
if (n <= 2)
return n;
int a = 1;
int b = 2;
int result = 0;
for (int i = 3; i <= n; i++) {
result = a + b;
a = b;
b = result;
}
return result;
}
}