题目描述
难度:简单
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
提示:
0 <= n <= 100
分析
仔细分析就会发现,这题本质上和斐波那契数列一致,只不过是边界条件不同,我们可以利用类似斐波那契数列的做法来完成,也就是后一个数等于前两个数相加;
由于最近学了一手动态规划,决定利用这个简单的题试试水,谁不爱一只会动态规划的小青蛙呢?
定义状态 dp[i] 为第 i 级台阶全部跳法,定义初始值 dp[0] dp[1] 为 1;
因为跳 1 阶到达终点是一种跳法 ,跳 2 阶到达终点也是一种跳法,总共就这两种跳法能跳到终点,两种跳法加起来就是到达终点的全部跳法;
所以有状态转移方程 dp[i] = dp[i - 1] + dp[i - 2];
代码:
/*
定义状态 dp[i] 是 i 个台阶会有多少种跳法
边界条件: dp[0] = 1 dp[1] = 1
状态转移方程 dp[i] = dp[i-1] + dp[i - 2]
因为跳 1 步到达终点是一种跳法 跳 2 步到达终点也是一种跳法 总共就这两种跳法 加起来就是总共的跳法了
*/
public static int numWays(int n) {
if(n == 0 || n == 1){ // 边界直接返回
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]) % 1000000007; // 由于题目限制,所以每次的跳法都求模
}
return dp[n];
}
总结
利用动态规划写这题效率挺高的,再接再厉!
岁月悠悠,衰微只及肌肤;热忱抛却,颓废必致灵魂