【LeetCode】第1269题——停在原地的方案数（难度：困难）

题目描述

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

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

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

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

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

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

3. 示例 3：
   输入：steps = 4, arrLen = 2
   输出：8

提示：
1 <= steps <= 500
1 <= arrLen <= 10⁶

来源：力扣（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]（上一步左移一次）

当然上述公式需要考虑越界问题。

由于只使用上一行dp，也可把dp数组压缩成两行，用时间换空间。

而且由于移动至太靠右边可能无法移回下标为0，所以dp数组的列数可以为Math.min(steps/2+1, arrLen)

代码详解

class Solution {
    public int numWays(int steps, int arrLen) {
        int col = Math.min(steps/2+1, arrLen); // 压缩列数
        int[][] dp = new int[steps+1][col];
        dp[0][0] = 1; // 这个赋值很灵魂
        // dp
        for(int i = 1; i <= steps; ++i) {
            for(int j = 0; j < col; ++j) {
                dp[i][j] = dp[i-1][j];
                if(j > 0) {
                    dp[i][j] = (dp[i][j] + dp[i-1][j-1]) % 1000000007;
                }
                if(j < col - 1) {
                    dp[i][j] = (dp[i][j] + dp[i-1][j+1]) % 1000000007;
                }
            }
        }
        return dp[steps][0];
    }
}

注意点

1. 也可把dp数组压缩成两行，用时间换空间
2. dp数组的列数不一定为arrLen，取Math.min(steps/2+1, arrLen)是最优的
3. 这道题可能一开始想成树的结构然后用递归，递归写起来很简单，但是这样时间复杂度太大，会超时