题目:青蛙跳台阶
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:输入:n = 2; 输出:2
示例 2:输入:n = 7; 输出:21
示例 3:输入:n = 0; 输出:1
提示:0 <= n <= 100
算法思路:
设跳上n级台阶有f(n)种跳法。在所有跳法中,青蛙的最后一步只有两种情况: 跳上1级或2级台阶。
当为1级台阶: 剩 n-1个台阶,此情况共有 f(n-1)种跳法;
当为2级台阶: 剩 n-2个台阶,此情况共有 f(n-2)种跳法。
f(n)为以上两种情况之和,即 f(n)=f(n-1)+f(n-2),以上递推性质为斐波那契数列。本题可转化为求斐波那契数列第 n项的值 ,与斐波那契数列 等价,唯一的不同在于起始数字不同。
青蛙跳台阶问题: f(0)=1, f(1)=1 , f(2)=2 ;
斐波那契数列问题: f(0)=0f , f(1)=1f, f(2)=1。
代码实现:
#include<iostream>
using namespace std;
int numWays(int);
int main(){
int pr=numWays(7);
cout<<"方式:"<<pr<<endl;
return 0;
}
int numWays(int n) {
if(n == 0 || n == 1){ return 1;}
return (numWays(n-1) + numWays(n-2))% 1000000007;
}
时间复杂度O(n)
优化一:记忆递归
但递归由于重复计算,导致效率很差,有很多重复的计算,相同的颜色表示计算多次,因此采用记忆递归
优化二:动态规划
#include<iostream>
using namespace std;
void numWays(int n);
int main() {
numWays(5);
}
int numWays(int n) {
//动态规划4行解决
//声明n+1大小的vector,+1是要存储0至n共n+1个数据。
vector<int> v(n + 1, 1);
for(int i = 2; i <= n; i++){
v[i] = (v[i - 1] + v[i - 2]) % 1000000007; //注意别忘记取余
return v[n]; //直接返回最后一个数,即最终结果
}
}