leetcode 70.爬楼梯 Java

题目链接

https://leetcode-cn.com/problems/climbing-stairs/

描述

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

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

注意：给定 n 是一个正整数。

示例

示例1

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

示例2

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

代码

简单的动态规划思想，当前的楼梯层只能由前一层或者前两层达到，因此

class Solution {
    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) {
       int prev1 = 0;
       int prev2 = 1;
       int cur = 0;

       for (int i = 1; i <= n; i++) {
           cur = prev1 + prev2;
           prev1 = prev2;
           prev2 = cur;
       }

       return cur;
    }
}