【Java】青蛙跳台阶问题（递归+备忘录）

题目描述

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

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

示例 1：

输入：n = 2
输出：2

示例 2：

输入：n = 7
输出：21
提示：

0 <= n <= 100

解题思路

很自然想到递归，但是开销很大，时间复杂度O(n)，空间复杂度O(n)，而且会超时，

可以考虑加入备忘录，因为同时递归的时候会出现重复的n，这时候可以先在HashMap中记录下来，

譬如当n=5的时候，需要递归计算n=4，n=3的情况，继续递归，计算n=4时需要分别计算n=3，n=2的递归，这时候就出现了重复计算，第一次计算的n=3的递归可以先存起来，下次就不用算了，原理就是这样。

class Solution {

    static int mod = 1000000007;
    public int numWays(int n) {
        HashMap<Integer,Integer> map = new HashMap<>();
        return helper(n,map);
    }

    public static int helper(int n,HashMap<Integer,Integer> map){
        if(n <= 1){
            return 1;
        }

        if(n==2){
            return 2;
        }

        if(map.containsKey(n)){
            return map.get(n);
        }else{
            int value = (helper(n-1, map) + helper(n-2, map))%mod;
            map.put(n, value);
            return value;
        } 
    }
}