题目描述:
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n
级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2 输出:2
示例 2:
输入:n = 7 输出:21
示例 3:
输入:n = 0 输出:1
提示:
0 <= n <= 100
方法:动态规划法
首先定义状态:dp[n]dp[n]表示n级台阶的跳法,随后需要确定转移方程
假设已知第n-1n−1级台阶的跳法,青蛙在n-1n−1级台阶上时只有如下图一种跳法,所以青蛙在n-1n−1级台阶上时跳法为dp[n-1]*1dp[n−1]∗1。
因为青蛙还可以一次跳两阶,当已知n-2n−2级的跳法时,青蛙在n-2n−2级台阶上时也只有如下图一种跳法(不能跳到n-1n−1阶上,因为此种跳法已被包含在上面n-1n−1的跳法当中),青蛙在n-2n−2级台阶上时增加新跳法为dp[i-2]*1dp[i−2]∗1种。
综上青蛙跳到第n级台阶上的跳法dp[n] = dp[n-1] *1+ dp[n-2] *1dp[n]=dp[n−1]∗1+dp[n−2]∗1,即为动态规划中的状态转移方程。
- 状态定义:dp[n]dp[n]表示n级台阶的跳法
- 状态转移方程:dp[n] = dp[n-1] + dp[n-2]dp[n]=dp[n−1]+dp[n−2]
- 初始状态:dp[0] = 1,dp[1] = 1dp[0]=1,dp[1]=1
- 返回值:dp[]dp[]数组最后一个数
class Solution {
public 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;
}
}