青蛙跳台阶问题
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
解题思路:
和剑指offer刷题七是一样的思路,跳到三级台阶的方法数等于跳到二级的方法数加上跳到一级的方法数,以此类推。
java
class Solution {
public int numWays(int n) {
if(n == 0) return 1;
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 1;
for(int i = 2; i <= n; i++){
dp[i] = dp[i-1] + dp[i-2];
dp[i] %= 1000000007;
}
return dp[n];
}
}
python
class Solution(object):
def numWays(self, n):
"""
:type n: int
:rtype: int
"""
dic ={}
dic[0]=1
dic[1]=1
if n>1:
for i in range(2,n+1):
dic[i] =dic[i-1]+dic[i-2]
return dic[n]%1000000007