【多维 DP】力扣2400. 恰好移动 k 步到达某一位置的方法数目

给你两个 正 整数 startPos 和 endPos 。最初,你站在 无限 数轴上位置 startPos 处。在一步移动中,你可以向左或者向右移动一个位置。

给你一个正整数 k ,返回从 startPos 出发、恰好 移动 k 步并到达 endPos 的 不同 方法数目。由于答案可能会很大,返回对 109 + 7 取余 的结果。

如果所执行移动的顺序不完全相同,则认为两种方法不同。

注意:数轴包含负整数。

示例 1:

输入:startPos = 1, endPos = 2, k = 3

输出:3

解释:存在 3 种从 1 到 2 且恰好移动 3 步的方法:

  • 1 -> 2 -> 3 -> 2.
  • 1 -> 2 -> 1 -> 2.
  • 1 -> 0 -> 1 -> 2.
    可以证明不存在其他方法,所以返回 3 。

示例 2:

输入:startPos = 2, endPos = 5, k = 10

输出:0

解释:不存在从 2 到 5 且恰好移动 10 步的方法。

提示:

1 <= startPos, endPos, k <= 1000

动态规划

cpp 复制代码
class Solution {
public:
    int numberOfWays(int startPos, int endPos, int k) {
        int MOD = 1e9+7;
        int offset = 1000;
        int maxRange = startPos + k;
        int minRange = startPos - k;
        vector<vector<int>> dp(3 * offset + 2, vector<int>(k+1));
        dp[startPos + offset][0] = 1;

        for(int j = 1; j <= k; j++){
            for(int i = minRange + offset; i <= maxRange + offset; i++){
                dp[i][j] = (dp[i-1][j-1] + dp[i+1][j-1]) % MOD;
            }
        }
        return dp[endPos + offset][k];
    }
};

这道题我们定义了一个偏移量offset,offset用于处理负轴上的数据,使他们在dp上的索引是正数。这道题我们需要注意的是如何定义dp的范围,由于startPos和k都在1000以内,那么负轴上最低的是startPos-k,因为startPos是正整数,所以负轴上最低是-999,正轴上最高是startPos+k,也就是2000,那么他们的范围就是3000,由于方便处理动态规划过程中的边界问题,令dp的行大小为3000+2。

我们定义一个二维数组dp[i][j],代表当前位置i和已经走了j步的情况下,有几种情况。那么我们可以知道dp[i][j]可以由dp[i-1][j-1]和dp[i+1][j-1]状态转移而来。

最后我们返回dp[endPos+offset][k],也就是在终点并且已经走了k步的情况下的情况有多少种。

相关推荐
long3162 分钟前
K‘ 未排序数组中的最小/最大元素 |期望线性时间
java·算法·排序算法·springboot·sorting algorithm
进击的小头5 分钟前
FIR滤波器实战:音频信号降噪
c语言·python·算法·音视频
xqqxqxxq10 分钟前
洛谷算法1-1 模拟与高精度(NOIP经典真题解析)java(持续更新)
java·开发语言·算法
razelan10 分钟前
初级算法技巧 4
算法
砍树+c+v11 分钟前
3a 感知机训练过程示例(手算拆解,代码实现)
人工智能·算法·机器学习
zy_destiny12 分钟前
【工业场景】用YOLOv26实现4种输电线隐患检测
人工智能·深度学习·算法·yolo·机器学习·计算机视觉·输电线隐患识别
智驱力人工智能22 分钟前
货车违规变道检测 高速公路安全治理的工程实践 货车变道检测 高速公路货车违规变道抓拍系统 城市快速路货车压实线识别方案
人工智能·opencv·算法·安全·yolo·目标检测·边缘计算
罗湖老棍子25 分钟前
【例9.18】合并石子(信息学奥赛一本通- P1274)从暴搜到区间 DP:石子合并的四种写法
算法·动态规划·区间dp·区间动态规划
2301_8107301034 分钟前
python第四次作业
数据结构·python·算法
adam_life37 分钟前
区间动态# P1880 [NOI1995] 石子合并】
算法