`第85篇 LeetCode剑指Offer动态规划(二)青蛙跳台阶问题
1.题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
提示:
0 <= n <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/qing-wa-tiao-tai-jie-wen-ti-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.动态规划的解题步骤
2.1.dp[i]的定义
对上题而言,我们举一个数字,如8,那么dp[8]就表示青蛙上8级台阶的跳法,那么dp[i]就表示青蛙上i级台阶的跳法。
2.2.递推式
对于此题,我们可以这样思考,青蛙要跳到 i 级是从什么地方跳来的,因为青蛙可以跳一级或者跳两级,因此跳到 i 级可能是从i - 1级跳来的,也可能是从i - 2级跳来的,因此跳到 i 级的跳法就是dp[i - 1] + dp[i - 2],递推式就是dp[i] = dp[i - 1] + dp[i - 2];
2.3.dp初始化和递归函数出口
我有一个疑惑,为什么跳上0级的台阶是1种跳法。。。。。。。哈哈
因为i==0或者i == 1时是不满足递推式的,所以我们要给定初始值dp[0] = 1,dp[1] = 1,而这些初始化的地方就是递归函数的出口。
2.4.递归函数
int dp(int n) {
if(n < 2){
return 1;//出口,初始化的地方
}
return (dp(n - 1) + dp(n - 2)) % 1000000007;
}
2.5.递归改成迭代
为什么改成迭代呢?且看图?
从递归树可以看出,递归做了很多重复的事情,时间复杂度是指数级的,因此改成迭代,使用数组把计算过的数据记录下来,然后往后需要用到时直接使用即可,而不是去重复计算。
int dp(int n) {
if(n < 2) {
return n;
}
vector<int> dp(n, 0);//不考虑0,考虑0的话要初始化成dp(n + 1, 0)
dp[0] = 1;
dp[1] = 2;
for(int i = 2;i < n;i++) {
dp[i] = (d[i - 1] + dp[i - 2]) % 1000000007;
}
return dp[i - 1];
}
2.6.迭代优化
因为计算dp(n),只需要知道dp(n-1)和dp(n-2)就好了,所以我们只需使用两个变量保存dp(n-1)、dp(n-2)即可
int dp(int n) {
int f1 = 1;
int f2 = 1;
for(int i = 2;i <= n;i++) {
int f3 = (f1 + f2) % 1000000007;
f1 = f2;
f2 = f3;
}
return f2;
}
3.结语
这还不是正题。看不出什么效果来,因为这个题肉眼就能看出递推式。