剑指Offer 10-II 青蛙跳台阶问题

413 阅读1分钟
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

示例 1:

输入:n = 2
输出:2
示例 2:

输入:n = 7
输出:21

/**如果第一次跳1级台阶,那么剩下n-1级台阶,跳法是f(n-1)
   如果第一次跳2级台阶,那么剩下n-2级台阶,跳法是f(n-2)
   所以 没有台阶或者一个台阶时,只有一种跳法
   大于1时,则为f(n) = f(n-1) + f(n-2) 

*/
class Solution {
    public int numWays(int n) {
        if(n == 0) return 1;
        int[] dp = new int[n + 1];
        dp[0] = 1;
        dp[1] = 1;
        for(int i = 2; i < n + 1; i++){
            dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007;
        }
        return dp[n];
    }
}

image.png

参考:leetcode-cn.com/problems/qi…