剑指Offer 10- Ⅱ.青蛙跳台阶问题 C++

题目描述

一只青蛙一次可以跳上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

解题思路

入门级的动态规划问题
开一个dp数组，大小为n+1，dp[i]表示跳到i台阶有多少种方法
初始化:这几个特判掉就可以了
dp[0]=1
dp[1]=1
dp[2]=2
转移方程：dp[i]=dp[i-1]+dp[i-2]
因为如果要跳到第i台阶上，那么只有两种可能
可能1：从第i-1台阶跳过来的
可能2：从第i-2台阶跳过来的
所以我们要求第i台阶有多少种方法，只需要把第i-1台阶的方法+第i-2台阶的方法就可以了
注意题目中说的要对1000000007取模
最后返回dp[n]，即第n个台阶有多少种方法

代码

class Solution {
public:
    int numWays(int n) {
        if(n==0){//特判：n==0
            return 1;
        }
        if(n==1||n==2){//特判：n==1或者n==2
            return n;
        }
        int dp[n+1];//开数组，注意大小为n+1，含义为第i个台阶有多少中方法
        dp[0]=1;//初始化
        dp[1]=1;
        dp[2]=2;
        for(int i=3;i<=n;i++){//从第三个台阶开始遍历
            dp[i]=(dp[i-1]%1000000007+dp[i-2]%1000000007)%1000000007;//转移方程，注意根据题意取模
        }
        return dp[n];//最后返回dp[n]
    }
};