题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例:
输入:n=2
输出:2
输入:n=0
输出:1
输入:n=7
输出:21
思路:
表面上题目在描述一个跳台阶的青蛙,但其本质还是斐波那契数列的问题。从边界开始分析:
- n=1时,青蛙只有1种跳法
- n=2时,青蛙有2种跳法:(2) (1,1)
- n >= 2时,
n = 3时,青蛙有3种跳法,分别是(2,1) (1,2) (1,1,1)
这背后的规律就是 f(3) = f(2) + f(1); 即最后一步跳跃,是一步跳跃还是两步跳跃
则递推公式为 f(n) = f(n-1) + f(n-2)
又根据示例输出:f(0)=1
代码
//边界条件:f(1) = 1; f(2) = 2;
//递推公式:f(n) = f(n-1) + f(n-2);
//本质是斐波那契数列
class Solution {
public:
int numWays(int n) {
vector<int> fib;
fib.push_back(1);//fib[0]=1
fib.push_back(1);//fib[1]=1
fib.push_back(2);//fib[2]=2
for (int i = 3;i<=n;i++) {
fib.push_back((fib[i-1]+fib[i-2]) % 1000000007);
}
return fib[n];
}
};