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

冷酷的摸鱼小将

已于 2022-09-16 15:08:55 修改

阅读量225

点赞数

分类专栏：剑指 Offer 文章标签：链表 leetcode 数据结构

于 2021-11-22 11:17:48 首次发布

本文链接：https://blog.csdn.net/qq_42467009/article/details/121467593

版权

剑指 Offer 专栏收录该内容

119 篇文章 1 订阅

订阅专栏

这篇博客介绍了青蛙跳台阶问题的两种解决方案：动态规划和循环求余。动态规划方法通过建立dp数组，利用状态转移方程dp[i] = dp[i-1] + dp[i-2]求解；循环求余方案则通过循环更新a和b的值来得到结果。这两种方法均考虑了题目中阶数n对跳法的影响，并对结果进行了1e9+7的取模操作。

摘要由CSDN通过智能技术生成

题目

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

答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

示例

输入：n = 2
输出：2

输入：n = 7
输出：21

输入：n = 0
输出：1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/qing-wa-tiao-tai-jie-wen-ti-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：动态规划

在这里插入图片描述

Java实现

要跳到最后一级台阶：
有可能是倒数第二级跳一下过来，也可能是倒数第三级跳两下过来。

所以动态转移方程：
dp[i] = dp[i - 1] + dp[i - 2];

class Solution {
    public int numWays(int n) {
        if (n == 0) return 1;
        if (n <= 2) return n;
        int[] dp = new int[n + 1];
        dp[0] = 1;
        dp[1] = 1;
        dp[2] = 2;

        for (int i = 3; i <= n; i++) {
            dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007;
        }
        return dp[n];
    }
}

在这里插入图片描述

方法2：循环求余

Java实现

class Solution {
    public int numWays(int n) {
        int a = 1, b = 1, sum = 0;

        for (int i = 0; i < n; i++) {
            sum = (a + b) % 1000000007;
            a = b;
            b = sum;
        }
        return a;
    }
}