LeetCode 1269. 停在原地的方案数 (DP)

Posted loveyixuan

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode 1269. 停在原地的方案数 (DP)相关的知识,希望对你有一定的参考价值。

题目

有一个长度为 arrLen 的数组,开始有一个指针在索引 0 处。

每一步操作中,你可以将指针向左或向右移动 1 步,或者停在原地(指针不能被移动到数组范围外)。

给你两个整数 steps 和 arrLen ,请你计算并返回:在恰好执行 steps 次操作以后,指针仍然指向索引 0 处的方案数。

由于答案可能会很大,请返回方案数 模 10^9 + 7 后的结果。

示例1:

输入:steps = 3, arrLen = 2
输出:4
解释:3 步后,总共有 4 种不同的方法可以停在索引 0 处。
向右,向左,不动
不动,向右,向左
向右,不动,向左
不动,不动,不动

示例2:

输入:steps = 2, arrLen = 4
输出:2
解释:2 步后,总共有 2 种不同的方法可以停在索引 0 处。
向右,向左
不动,不动

示例3:

输入:steps = 4, arrLen = 2
输出:8

提示:

  • 1 <= steps <= 500
  • 1 <= arrLen <= 10^6

题解:

用DP求解,主要是状态转移。走i步到达j位置时,dp[i][j] = dp[i-1][j] + dp[i-1][j+1] + dp[i-1][j-1]。
注意起始位置和最远位置可以另外讨论,最远可以到达的位置 n = min(arrLen, steps)

代码:

class Solution:
    def numWays(self, steps: int, arrLen: int):
        
        n = min(arrLen, steps)
        mod = 10 ** 9 + 7
        dp = [[0 for i in range(n)] for i in range(steps+1)]
        
        dp[0][0] = 1  
        for i in range(1, steps+1):
            for j in range(n):
                if j==0:
                    dp[i][j] += dp[i-1][j] + dp[i-1][j+1]
                elif j==n-1:
                    dp[i][j] += dp[i-1][j] + dp[i-1][j-1]
                else:
                    dp[i][j] += dp[i-1][j] + dp[i-1][j+1] + dp[i-1][j-1]
                dp[i][j] %= mod
                
        return dp[steps][0]

以上是关于LeetCode 1269. 停在原地的方案数 (DP)的主要内容,如果未能解决你的问题,请参考以下文章

leetcode-164周赛-1269-停在原地的方案数

leetcode1269. 停在原地的方案数简单DP

LeetCode 1269. 停在原地的方案数 (Java 记忆化,动态规划)

LeetCode 1269. 停在原地的方案数 (Java 记忆化,动态规划)

我用java刷 leetcode 1269. 停在原地的方案数

LeetCode 1269 停在原地的方案数[动态规划] HERODING的LeetCode之路