一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
解:这道题和上一道斐波那契数列类似,所以我仍然采用递归调用:
class Solution {
public:
int numWays(int n) {
int count=1e9+7;
//1-1 2-2 3-3
//结束条件
if(n<=1) return 1;
//if(0<n<4) return n;
//等价函数
//如果第一步跳一个台阶
//int first=1;
int t1=(numWays(n-1))%count;
//如果第一步跳两个台阶
int t2=(numWays(n-2))%count;
return (t1+t2)%count;
}
};
虽然输出正确,但是依然出现超时的问题!
再看看一位大佬的超简解法:
class Solution {
public:
int numWays(int n) {
//创建一个数组,
//数组大小尽量大,因为它是可以取的n的最大范围
int ans[500]={1,1};
//把每一种台阶数对应的跳法总数存到对应数组中
for(int i=2;i<=n;i++) ans[i]=(ans[i-1]+ans[i-2])%(int(1e9+7));
//返回所取的台阶对应跳法数
return ans[n];
}
};
这种把每一个数值存进去的解法和上一个求斐波那契数列的解法非常相似,都是设定结束条件初始情况之后,利用循环把之后的每一种情况存到数组中去,然后需要哪个取哪个!这种方法对于那种与前一个数据和前前一个数据有关的数列非常有用!