动态规划-面试题08.01三步问题-力扣(LeetCode)

最新推荐文章于 2025-05-16 12:14:55 发布

1白天的黑夜1

最新推荐文章于 2025-05-16 12:14:55 发布

阅读量283

点赞数 6

文章标签：动态规划 leetcode 算法 c++

本文链接：https://blog.csdn.net/black_bear_white/article/details/147705678

版权

一、题目解析

此题可以类比第N个泰波那契数

二、算法解析

1、状态表示

根据上面的分析和题目要求，dp[i]表示：到达i位置，一共有多少种方法

2、状态转移方程

以i位置的状态，以最近一步划分问题

dp[i] 从i-1->i dp[i-1]

从i-2->i dp[i-2]

从i-3->i dp[i-3]

3、初始化

计算四节台阶的方法需要前三阶各自的方法数，所以初始化dp[1] = 1,dp[2] = 2,dp[3] = 4

4、填dp表的顺序

以4为起点，从左往右依次填写

5、返回值

根据题目要求需要返回在i点的方法数，所以返回dp[n]

小细节：

题目上说数字会过大所以需要模上一个10000000007，也就是1e9+7

老规矩，先去自己实现结合解析，链接：面试题 08.01. 三步问题 - 力扣（LeetCode）

三、代码示例

class Solution {
public:
    int waysToStep(int n) {
        const int MOD = 1e9 + 7;//定义模数
        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-1] + dp[i-2]) % MOD + dp[i-3]) % MOD;//对两个加法分别取模
        }
        return dp[n];//返回结果
    }
};