问题:一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
从简单的问题出发,如果只有一个台阶,则跳法f(1)=1,如果有两个台阶,则跳法f(2)=2。
如果有三个台阶,则在跳到第三个台阶之前有两种可能性,一种是青蛙处于第一个台阶上,下一步需要跳两个台阶到达第三个台阶;还有一种可能就是青蛙处于第二个台阶上,下一步只需跳一个台阶就可以到达第三个台阶,易知到达第一个台阶上的跳法有f(1)种,达到第二个台阶的跳法有f(2)种,所以说综上可得到达第三个台阶的跳法为f(3)=f(1) + f(2)。
同理可推出,如果有n个台阶,那么在跳到此台阶之前青蛙只有两种可能性,一种是青蛙处于第n-1个台阶,另一种可能性是青蛙处于第n-2个台阶,青蛙跳上第n-1个台阶的跳法为f(n-1),跳上第n-2个台阶的跳法为f(n-2),所以跳上第n个台阶的跳法
f(n)=f(n-1)+f(n-2)。
class Solution {
public int numWays(int n) {
int[] dp = new int[Math.max(n+1,3)];
dp[0] = 1;
dp[1] = 1;
dp[2] = 2;
for(int i = 3;i <= n; i++){
dp[i] = (dp[i-1] + dp[i-2])%1000000007;
}
return dp[n];
}
}
注:其中Math.max(n+1,3)的内部函数为一个三目表达式:n+1 > 3 ? n+1:3