剑指 Offer 10- II. 青蛙跳台阶问题

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

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

示例 1：

输入：n = 2
输出：2
示例 2：

输入：n = 7
输出：21
示例 3：

输入：n = 0
输出：1

题解

题目很明显是一个递归问题，可以将整体问题划分为多个小问题：

我们先假设楼梯只有一阶，也就是 n 为1；此时只有一种办法能爬到楼梯，也就是一次爬一阶楼梯。
我们再假设楼梯只有二阶，也就是 n 为2；此时有两种办法能爬到楼梯，一种是一次爬两阶楼梯，一种是爬两次一阶楼梯。

我们再来看楼梯只有三阶，也就是 n 为3；此时问题可以划分为多个子问题，也就是 第一步爬一阶楼梯的所有情（可以表示成 楼梯还剩下n-1的所有走法）加上 第一步爬二阶楼梯（可以表示成楼梯还剩下n-2 的所有走法）的所有情况。

如果用函数来表示以上文字的话。

代码如下：

class Solution {
    public int climbStairs(int n) {
        if(n==1){
            return 1;
        }       
        if(n==2){
            return 2;
        }
        return climbStairs(n-1)+climbStairs(n-2);
    }
}

我们提交以上代码发现出现超出时间限制的问题，原因是使用以上递归，虽然代码简单，但是进行了多次的重复计算。

进一步改进，使用 HashMap 来保存我们已经求解过的值。首先从 HashMap 中判断当前 f(n) 是否已经计算过，如果计算过，直接取出数据返回。如果 HashMap 中没有当前 f(n) 的值则将计算的值保存到 HashMap 中。

class Solution {
    private static Map<Integer,Integer> hashMap = new HashMap<>();
    public int numWays(int n) {
        
        if(n == 0){
            return 1;
        }
        if(n == 1){
            return 1;
        }
        if(n == 2){
            return 2;
        }
        if(hashMap.get(n) != null){
            return hashMap.get(n);
        }else{
            int ret = (numWays(n-1)+numWays(n-2))%1000000007;
            hashMap.put(n,ret);
            return ret;
        }
    }
}