青蛙跳台阶问题
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
做法:青蛙来到最后一步有两种情况:跳1阶,此时对应有f(n-1)种方法;跳2阶,此时对应有f(n-2)种方法。即f(n) = f(n-1) + f(n-2),为斐波那契数列问题,只是初始值不同而已
class Solution {
public int numWays(int n) {
if(n == 0){
return 1;
}
if(n == 1){
return 1;
}
if(n == 2){
return 2;
}
//f(n) == (f(n-1) + 1) + (f(n-2) + 2)
int[] dp = new int[n+1];
dp[0] = 1;
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i < dp.length; i++){
dp[i] = dp[i-2] + dp[i-1];
dp[i] %= 1000000007;
}
return dp[n];
}
}