题目描述
一只青蛙一次可以跳上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;
}
}
}