剑指offer:青蛙跳台问题
问题描述:一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
解题思路:
本题可以借鉴斐波那契数列的思想:n个台阶有f(n)种跳法,当青蛙站在第一节台阶上时,则对于原来的n个台阶的有f(n-1)种跳法,以此类推f(n)=f(n-1)+f(n-2),f(n-1)=f(n-2)+f(n-3)…其中f(0)=1,f(1)=1{在斐波那契数列中f(0)=0,f(1)=1}。根据该思路可以得到代码如下:
C++:
class Solution
{
int numWays(int n)
{
int a=1,b=1,sum;
for(int i=0;i<n;i++)
{
sum=(a+b)%1000000007;
a=b;
b=sum;
}
return a;
}
};
Python:
class Solution:
def numWays(self,n):
a=1
b=1
for i in range(n):
a,b=b,a+b
return a%1000000007