70. Climbing Stairs

70. 爬楼梯

那么首先使用动态规划的典型题目来演示一下动态规划的操作。

题目描述

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。

每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？

示例 1：

输入： 2
输出： 2
解释： 有两种方法可以爬到楼顶。
1.  1 阶 + 1 阶
2.  2 阶

示例 2：

输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶

题解

分析一下问题，每次可以爬1到2个台阶，也就是说，对于第i个台阶，我们可以通过第i-1个或者第i-2个台阶得到。

那么我们就可以得到状态转移方程：dp[i]= dp[i-1]+dp[i-2];

边界条件：我们需要考虑dp[0]和dp[1]设置为1.

• 从第 0 级开始爬的，所以从第 0 级爬到第 0 级我们可以看作只有一种方案，即 f(0)=1；
• 从第 0 级到第 1 级也只有一种方案，即爬一级，f(1)=1。

优化 ：压缩空间，由于只需要最后的n，那么可以使用一个空间来保存最终数据，不需要进行中间过程的保存。

代码

class Solution {
    public int climbStairs(int n) {
        if(n <= 2){
            return 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];
    }
}

优化方法：

class Solution {
    public int climbStairs(int n) {
        if(n <= 2){
            return n;
        }
        int prepre = 1;
        int pre = 2;
        int cur =0;
        for(int i=3; i<=n; i++){
            cur = prepre + pre ;
            prepre = pre;
            pre =  cur;
        }
        return cur;
    }
}