LeetCode 1269. 停在原地的方案数--动态规划

1269. 停在原地的方案数

有一个长度为 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 <= 10e6

题解

一开始想复杂了，想到排列组合去了，从此便一去不复返，，，，后来看数据量比较小，考虑到应该可以动态规划推导，思考了下确实如此，虽然数组长度最大为10e6，但是受steps影响最多也只能到500，于是变成了500x500的动态规划。

我们定义dp[i][j]表示恰好移动了i次到达下标j的方案数，那么转移关系如下：

if(j-1>=0)
dp[i][j]+=dp[i-1][j-1];//上一个状态往右走了一步
dp[i][j]+=dp[i-1][j];//上一个状态留在原地
if(j+1<=min(steps,arrLen-1))
dp[i][j]+=dp[i-1][j+1];//上一个状态往左走了一步

最后返回
dp[steps][0]即可。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod=1000000007;
    ll dp[510][510];
    int numWays(int steps, int arrLen) {
        int n=min(steps,arrLen-1);//能到达的最大下标
        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        for(int i=1;i<=steps;i++)
        {
            for(int j=0;j<=n;j++)
            {
                dp[i][j]+=dp[i-1][j];
                if(j-1>=0)
                dp[i][j]+=dp[i-1][j-1];
                if(j+1<=n)
                dp[i][j]+=dp[i-1][j+1];
                dp[i][j]%=mod;
            }
        }
        return int(dp[steps][0]);
    }
};