leetcode专题训练1269. Number of Ways to Stay in the Same Place After Some Steps

最新推荐文章于 2023-11-28 14:37:07 发布

Emma1997

最新推荐文章于 2023-11-28 14:37:07 发布

阅读量205

点赞数

分类专栏： leetcode+剑指offer

本文链接：https://blog.youkuaiyun.com/Ema1997/article/details/116762988

版权

leetcode+剑指offer 专栏收录该内容

135 篇文章

订阅专栏

本文介绍了LeetCode第1269题的解决方案，通过动态规划的方法求解停留在原地的步骤数。原题考虑使用滚动数组优化，但最终采用二维数组实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

又是一道dp，本来想着滚动数组做，后来想了想好像不太好写，就直接开了个二维数组。

class Solution:
    def numWays(self, steps: int, arrLen: int) -> int:
        maxNum = 10**9+7
        dp = [[0 for i in range(min(steps+1, arrLen))] for j in range(steps+1)]
        dp[0][0] = 1

        for i in range(1, steps+1):
            for j in range(min(steps+1, arrLen)):
                # print(i, j)
                # 停留在原地
                dp[i][j] = dp[i-1][j]
                # 从左边转移
                if j-1 >= 0:
                    dp[i][j] = (dp[i][j]+dp[i-1][j-1])%maxNum
                # 从右边转移
                # 最多只能到steps+1的位置，不然肯定回不去了
                if j+1 < min(steps+1, arrLen):
                    dp[i][j] = (dp[i][j]+dp[i-1][j+1])%maxNum

        return dp[steps][0]