LeetCode每日抑题 1269. 停在原地的方案数

1269. 停在原地的方案数

难度：困难
语言：java

题目内容

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

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

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

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

解题思路

其实这是一个比较明显的dp题目，下一步的走向，由剩下的步数，以及所在的位置决定，但是分以下的几种情况

1. 剩下的步数，大于已经向右的步数，那这个时候有三种选择方案，左右不动
2. 剩下的步数，等于已经向右的步数，那这个时候只有一种方案，就是立刻回头
3. 如果所在的位置是最右的位置，只可以向左或者向右
4. 如果所在的位置是最左的位置，只可以向右或者向左

很明显就可以设置dp的数组为两项，一个是走过的步数，一个是所在的位置。
但是我在写代码的时候发现1、2两点其实是一个意思，因为对一个点来说，只要是可行的，都是由三个方向决定的，所以我将1、2两点合并

class Solution {
    public int numWays(int steps, int arrLen) {
        int[][] dp = new int [steps+1][arrLen];
        dp[0][0]=1;
        for(int i =1 ; i<=steps;i++){
            for (int j = 0;j<arrLen;j++){
                if (j==arrLen-1){
                    dp[i][j] = dp[i-1][j-1]+dp[i-1][j];
                }
                else if (j == 0){
                    dp[i][j] = dp[i-1][j]+dp[i-1][j+1];
                }
                else{
                    dp[i][j] = dp[i-1][j-1]+dp[i-1][j]+dp[i-1][j+1];
                }
            }
        }
        for (int i = 0;i<dp.length;i++){
            System.out.println(Arrays.toString(dp[i]));
        }
        return dp[steps][0];
    }
}

答案里我没取模，对比一下答案
跟答案相比，有两个点我处理的不太好，第一个点是数组的右边界，应该取步数或者右边界的最小值

class Solution {
    public int numWays(int steps, int arrLen) {
        final int MODULO = 1000000007;
        int maxColumn = Math.min(arrLen - 1, steps);
        int[][] dp = new int[steps + 1][maxColumn + 1];
        dp[0][0] = 1;
        for (int i = 1; i <= steps; i++) {
            for (int j = 0; j <= maxColumn; j++) {
                dp[i][j] = dp[i - 1][j];
                if (j - 1 >= 0) {
                    dp[i][j] = (dp[i][j] + dp[i - 1][j - 1]) % MODULO;
                }
                if (j + 1 <= maxColumn) {
                    dp[i][j] = (dp[i][j] + dp[i - 1][j + 1]) % MODULO;
                }
            }
        }
        return dp[steps][0];
    }
}