动态规划之机器人到达指定位置的方法数

机器人到达指定位置的方法数

链接：https://www.nowcoder.com/questionTerminal/54679e44604f44d48d1bcadb1fe6eb61
来源：牛客网

假设有排成一行的N个位置，记为1~N，开始时机器人在M位置，机器人可以往左或者往右走，如果机器人在1位置，那么下一步机器人只能走到2位置，如果机器人在N位置，那么下一步机器人只能走到N-1位置。规定机器人只能走k步，最终能来到P位置的方法有多少种。由于方案数可能比较大，所以答案需要对1e9+7取模。

输入描述:
输出包括一行四个正整数N（2<=N<=5000）、M(1<=M<=N)、K(1<=K<=5000)、P(1<=P<=N)。
输出描述:
输出一个整数，代表最终走到P的方法数对1e9+7取模后的值。
时间复杂度O(n*k),空间复杂度O(N)。

n, m, k, p = list(map(int, input().split()))
eps = 1e9 + 7

last = [0] * (n + 2)
curr = [0] * (n + 2)
last[m] = 1
for i in range(1, k+1):
    l = max(1, m-i)
    r = min(n, m+i)
    for j in range(l, r+1):
        curr[j] = (last[j-1] + last[j + 1]) % eps
    last[:] = curr[:]
print(int(curr[p]))