力扣面试题 08.01. 三步问题

解决三步问题：动态规划算法求楼梯上楼方式总数

最新推荐文章于 2025-03-24 08:28:57 发布

原创最新推荐文章于 2025-03-24 08:28:57 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #动态规划 #算法

面试题专栏收录该内容

52 篇文章

订阅专栏

本文介绍如何使用动态规划方法解决LeetCode题目——三步楼梯问题，计算小孩上n阶台阶的不同走法。通过Java实现并处理结果溢出，给出代码示例和结果解释。

题目

三步问题。有个小孩正在上楼梯，楼梯有n阶台阶，小孩一次可以上1阶、2阶或3阶。实现一种方法，计算小孩有多少种上楼梯的方式。结果可能很大，你需要对结果模1000000007。

示例

输入：n = 3
输出：4
说明: 有四种走法

输入：n = 5
输出：13

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/three-steps-problem-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：动态规划

注意溢出，dp数组取 long型

Java实现

class Solution {
    public int waysToStep(int n) {
        //下标从0开始，所以申请n+1个空间
        if (n == 1) return 1;
        if (n == 2) return 2;
        if (n == 3) return 4;

        long[] dp = new long[n + 1];
        dp[0] = 0;
        dp[1]= 1;
        dp[2] = 2;
        dp[3] = 4;

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