剑指 Offer 10- II. 青蛙跳台阶问题
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
动态规划
跟上题一样,只有初始状态不一样
大数越界:随着随着 n 增大, f(n) 会超过 Int32 甚至 Int64 的取值范围,导致最终的返回值错误。
- 求余运算规则: 设正整数 x,y,p ,求余符号为⊙,则有 (x+y)⊙p=(x⊙p+y⊙p)⊙p 。
- 解析: 根据以上规则,可推出 f(n)⊙p=[f(n−1)⊙p+f(n−2)⊙p]⊙p ,从而可以在循环过程中每次计算
sum=a+b⊙1000000007 ,此操作与最终返回前取余等价。
class Solution {
public:
int numWays(int n) {
vector<int>dp;
for(int i=0;i<=n;i++)
{
if(i == 0)
{
dp.push_back(1);
}
else if(i == 1)
{
dp.push_back(1);
}
else if(i == 2)
{
dp.push_back(2);
}
else
{
dp.push_back((dp[i-1]+dp[i-2])%1000000007);
}
}
return dp[n];
}
};
同样可以降低空间复杂度