剑指Offer算法题——青蛙跳台阶问题（JAVA）

题目描述：

        一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

        答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

示例 1：

输入：n = 2
输出：2

示例 2：

输入：n = 7
输出：21

示例 3：

输入：n = 0
输出：1

分析：

        我们正向思考问题，一次跳1阶或2阶，这样分析问题，每次都要分成两个分支。我们不如反着来思考：

         根据这个思路，假设我们要4阶楼梯，上第4阶楼梯之前只有两种情况：在第3个台阶或在第2个台阶，只要把最后跳1阶即（前面跳第3阶楼梯的方法数）加上最后跳2阶（前面跳第2阶楼梯的方法数）就可以啦！

即跳n阶楼梯的方法数=跳n-1阶楼梯的方法数+跳n-2阶楼梯的方法数

 到这里，这和我上次写的斐波那契数列一样！f(n)=f(n-1)+f(n-2)

 好家伙，代码如下：（取余1000000007是为了防止溢出）

public class Solution {
    public int numWays(int n) {
        //台阶为0或1时，只有一种方法
        if (n == 0 || n == 1) {
            return 1;
        }
        int[] dp = new int[n + 1];
        dp[1] = 1;
        dp[2] = 2;
        for (int i = 3; i <= n; i++) {
            dp[i] = (dp[i - 2] + dp[i - 1]) % 1000_000_007;
        }
        return dp[n];
    }
}

我们来优化一下，进行空间压缩

        其实我们不必把所有的记录都记起来，假设在4阶楼梯，我们只需知道跳2阶和跳3阶的方法是多少即可，所以每次就保留n-1阶和n-2阶方法数就好了。

public class Solution {
    public int numWays(int n) {
        if (n == 0 || n == 1) {
            return 1;
        }
        int pre = 1, cur = 2;
        for (int i = 3; i <= n; i++) {
            int tmp = (pre + cur) % 1000_000_007;
            pre = cur;
            cur = tmp;
        }
        return cur;
    }
}

        循坏中先使用变量tmp记录n-1阶和n-2阶的方法数，再将n-2阶的方法数用n-1阶的方法数替换，将n-1阶的方法数用最新的tmp替换，以此形成递归。