三步问题 | 动态规划

1.三步问题

题目连接:面试题 08.01. 三步问题

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

2.题目解读

第1阶:小孩跨1阶,就1种方式

第2阶:小孩从第0阶,跨2阶,1种方式。在第1阶的基础上跨1阶,就上去了。加起来2种方式

第3阶:小孩从第0阶,跨3阶,在第1阶的基础上跨2阶,在第2阶的基础上跨1阶。加起来4种方式

可以找到规律,0-3阶是特殊情况,往后的,是在原来的基础(原来阶段的结果)上跨1、2或3步,走到n阶位置。说白了:n > 3时,上n阶台阶的方式就是前3个的和。

3.解决问题

(1)、状态表示

dp[i] 表⽰:到达 i 位置时,⼀共有多少种⽅法。
(2)、状态转移⽅程

dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3]

注意的是,dp[i - 1] + dp[i - 2] + dp[i - 3] 取模会溢出:signed integer overflow。两个数相加/乘,都需要取⼀次模。
(3)、初始化

dp[1] = 1, dp[2] = 2, dp[3] = 4
(4)、初始化顺序

这里根据题目要求,根据状态转移方程,从左往右,将dp填充。
(5)、返回值

返回dp[n]即为,返回结果。

4.参考代码(非空间优化版本)

C++版本:

cpp 复制代码
class Solution {
public:
    const int MOD = 1e9 + 7;
    int waysToStep(int n) 
    {
        if(n == 0) return 0;
        if(n == 1 || n == 2) return n;
        if(n == 3) return 4;

        vector<int> dp(n + 1);
        dp[1] = 1;
        dp[2] = 2;
        dp[3] = 4;

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

        return dp[n];
    }
};

Java版本:

java 复制代码
class Solution {
    final int MOD = 1000000007;

    public int waysToStep(int n) {
        if(n == 0) return 0;
        if(n == 1 || n== 2) return n;
        if(n == 3) return 4;

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

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

        return dp[n];
    }
}
相关推荐
卷福同学1 小时前
【AI编程】AI+高德MCP不到10分钟搞定上海三日游
人工智能·算法·程序员
mit6.8241 小时前
[Leetcode] 预处理 | 多叉树bfs | 格雷编码 | static_cast | 矩阵对角线
算法
皮卡蛋炒饭.2 小时前
数据结构—排序
数据结构·算法·排序算法
??tobenewyorker2 小时前
力扣打卡第23天 二叉搜索树中的众数
数据结构·算法·leetcode
贝塔西塔3 小时前
一文读懂动态规划:多种经典问题和思路
算法·leetcode·动态规划
众链网络3 小时前
AI进化论08:机器学习的崛起——数据和算法的“二人转”,AI“闷声发大财”
人工智能·算法·机器学习
3 小时前
Unity开发中常用的洗牌算法
java·算法·unity·游戏引擎·游戏开发
飒飒真编程5 小时前
C++类模板继承部分知识及测试代码
开发语言·c++·算法
GeminiGlory5 小时前
算法练习6-大数乘法(高精度乘法)
算法
熬了夜的程序员5 小时前
【华为机试】HJ61 放苹果
算法·华为·面试·golang