剑指 Offer 10- II. 青蛙跳台阶问题
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
要求第 n n n阶台阶有几种上法,即 f ( n ) f(n) f(n),可以分解为,先上到第 n − 1 n-1 n−1阶,然后再上一阶;或者先上到 n − 2 n-2 n−2阶,再上2阶。即有递推公式: f ( n ) = f ( n − 1 ) + f ( n − 2 ) f(n) =f(n-1)+f(n-2) f(n)=f(n−1)+f(n−2)
一眼可以看出这是 斐 波 那 契 函 数 斐波那契函数 斐波那契函数的递推公式,只是初始值有所区别。
斐波那契:f(0)=0 f(1)=1 f(2)=1
本问题 :f(0)=1 f(1)=1 f(2)=2
故,参考剑指 Offer 10- I. 斐波那契数列解法二,代码如下:
class Solution {
public int numWays(int n) {
final int MOD = 1000000007;
int p=0,q=1,r=1;
for(int i=0;i<n;i++){
p = q;
q = r;
r = (p+q)%MOD;
}
return q;
}
}