有一个长度为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-ways-to-stay-in-the-same-place-after-some-steps
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
这题一看就是DP, 因为当前状态可以由上一步状态进行三种操作得到。
定义dp[i][j]为 走 i 步,到 位置 j 的解。
一般而言, dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j] + dp[i - 1][j + 1],
三项分别代表,从上一步状态向右走;保持不动;向左走。
然后需要注意,可以走的最远距离是min(steps, arrLen),避免dp数组开得太大导致计算太多超时。
时间复杂度:O(M * min(M,N)), M = len(steps), N = len(arrLen)
空间复杂度:O(M * min (M, N))
class Solution(object):
def numWays(self, steps, arrLen):
"""
:type steps: int
:type arrLen: int
:rtype: int
"""
# dp[i][j] 代表走 i 步,到位置 j 的解
# dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j] + dp[i - 1][j + 1]
n = min(steps, arrLen)
dp = [[0 for _ in range(n)] for _ in range(steps + 1)]
mod = 10 ** 9 + 7
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][0] + dp[i - 1][1]) % mod
elif j == n - 1:
dp[i][j] += (dp[i - 1][j] + dp[i - 1][j - 1]) % mod
else:
dp[i][j] += (dp[i - 1][j - 1] + dp[i - 1][j] + dp[i - 1][j + 1]) % mod
return dp[steps][0]