leet code 面试题 08.01. 三步问题练习

soEasyp

已于 2022-02-23 20:15:28 修改

阅读量94

点赞数

分类专栏：面试文章标签：算法

于 2022-02-22 11:35:07 首次发布

本文链接：https://blog.csdn.net/sinat_34709499/article/details/123064674

版权

面试专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该博客介绍了一个使用动态规划算法解决的问题：小孩上楼梯有多种方式。当楼梯阶数为n时，小孩可以每次上1、2或3阶，求所有可能的上楼方式总数。博客提供了一个Java实现的解决方案，通过一个数组记录每个阶数的上楼方式，并利用前两阶和前三阶的上楼方式计算当前阶数的方式数，结果对1000000007取模。示例展示了对于n=3和5的输入，其输出分别为4和13。

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

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

示例1:

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

输入：n = 5
输出：13
提示:

n范围在[1, 1000000]之间

class Solution {
    public int waysToStep(int n) {
        if (n < 3) {
           return n;
        }
        if (n == 3) {
           return 4;
        }
        int[] dip = new int[n+1];
        dip[1] = 1;
        dip[2] = 2;
        dip[3] = 4;
        for(int i = 4; i < n + 1; i++) {
           dip[i] = ((dip[i-1] + dip[i-2])% 1000000007 + dip[i-3])%1000000007;
        }
        return dip[n];
    }
}