Leecode笔记-1269. 停在原地的方案数

@1269. 停在原地的方案数

有一个长度为 arrLen 的数组，开始有一个指针在索引 0 处。
每一步操作中，你可以将指针向左或向右移动 1 步，或者停在原地（指针不能被移动到数组范围外）。
给你两个整数 steps 和 arrLen ，请你计算并返回：在恰好执行 steps 次操作以后，指针仍然指向索引 0 处的方案数。
由于答案可能会很大，请返回方案数 模 10^9 + 7 后的结果。

思路

动态规划：用 dp[i][j] 表示在第i步操作之后，指针位于j的方案数。经过一步到达j的方法有：

• 由「原地」操作到达当前状态，即由dp[i-1][j]转移而来
• 由「向右」操作到达当前状态，即由dp[i-1][j-1]转移而来
• 由「向左」操作到达当前状态，即由dp[i-1][j+1]转移而来
  总的方案数为三者相加的值。
  一步不走时，指针位于0的位置，设置初始状态dp[0][0] = 1;
  数组大小[steps+1, min(steps+1, arrLen)] ， i, j范围为step，min(steps， arrLen-1)；
  要注意边界的情况当: j=0时，dp[i-1][j-1]=0; j为dp最后一个数时，dp[i-1][j+1]=0；

优化

数组维度：二维dp会耗费更多内存空间，而dp[i][j]只与上一步的三个状态有关系，所以可以将二维dp数组改为一维dp数组。
数组长度：走step步且能回到原点能到达的最大长度为step/2,数组的大小设置为min(step/2+1,arrLen)。

class Solution {
public:
    const int MODULO = 1000000007;
    int numWays(int steps, int arrLen) {
        if (arrLen == 1){
            return 1;
        }
        int maxcolumn = min(steps/2+1, arrLen);
        vector<int> dp(maxcolumn,0);
        dp[0] = 1;
        int temp = 0;
        for(int step = 1; step <= steps; step++){
            int minnum = min(step,maxcolumn);
            temp = dp[minnum];
            //分成三种情况计算dp[j]
            for(int j = minnum; j >= 0; j--){
                if(j == maxcolumn){
                    temp = dp[j];
                    dp[j] = (dp[j] + dp[j-1]) % MODULO;
                }else if(j == 0){
                    dp[j] =(dp[j] + temp)% MODULO;
                    //cout<<j<<" "<< dp[j]<<" ";
                }else{
                    int temp1 = dp[j];
                    dp[j] = ((dp[j-1] + dp[j])% MODULO + temp) % MODULO;
                    temp = temp1;
                }
            }
        }
        return dp[0];
    }
};

//计算dp代码改进
for(int step = 1; step <= steps; step++){
            vector<int> dp1(mincolum,0);
            int edge = min(step,mincolum-1);//可到达的最远的位置
            for(int j = 0; j <= edge; j++){
                dp1[j] = dp[j];
                
                if(j-1>=0){
                    dp1[j] = (dp1[j] + dp[j-1])%MODULO;
                    //cout<<dp1[j]<<" ";
                }
                if(j+1<=mincolum-1) {
                    dp1[j] = (dp1[j]+dp[j+1])%MODULO；
                }
            }
            dp = dp1;
        }