【leetcode】停留在原地的方案数(dp)

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

链接：https://leetcode-cn.com/problems/number-of-ways-to-stay-in-the-same-place-after-some-steps

思路分析：

一眼看过去就知道是dp了。但是我还是debug了半个多小时…
第一步，分析状态，使得该状态能储存所有情况的枚举。
这道题的状态很明显了。dp[i][j]表示通过 j 步走到第 i 个位置的方案数，我们要求的是dp[steps][0].
第二步，根据状态的决策写出状态转移方程。此题状态决策为不动，向左，向右。
状态转移方程也很好写：

1. dp[i][j] += dp[i][j-1].
   选择不动。
2. i > 0, dp[i][j] += dp[i-1][j-1].
   如果位置在起点右边，就可以向左移动。
3. i < arrLen, dp[i][j] += dp[i+1][j-1].
   如果位置不在右边界，就可以向右移动。

第三步，分析初始条件，从题意和所写的状态方程出发。
分析初始条件可以根据所写的状态转移方程找。比如说上面方程涉及到[j-1]，那么你就得初始化 j = 0的情况。j = 0 即0个方案数，很明显，只有dp[0][0] = 1, 其他情况均为0. 这就是它的初始条件。

第四步，写递推。
这一步，是我debug最难受的一步。因为我把两个for循环写反了，导致结果不正确。后来将样例输出出来，自己手动打dp表才发现问题…
我们第一层循环应该是步数，第二层循环是位置。我们得先等前n-1的所有位置情况都枚举完了，才能进行前n步。这和01背包的原理是一样的。我们得先等前n-1件物品选与不选的所有情况枚举完，才能在前n件物品这个状态中操作。

另外，这道题有一个坑，如果arrLen > steps, arrLen = steps+1. 因为从索引0走step步最多走到索引step，我们可以根据step的数据来减少我们的空间。

下面上代码：

const int MOD = 1e9+7;
class Solution {
public:
    int numWays(int steps, int arrLen) {
        int dp[505][505];
        memset(dp,0,sizeof(dp));
        if(arrLen > steps) arrLen = steps+1;
        dp[0][0] = 1;
        for(int j = 1;j <= steps;j++)
        {
            for(int i = 0;i < arrLen;i++)
            {
                dp[i][j] = dp[i][j-1];
                if(i > 0) dp[i][j] = (dp[i-1][j-1]+dp[i][j])%MOD;
                if(i < arrLen) dp[i][j] = (dp[i][j]+dp[i+1][j-1])%MOD;
                // dp[i][j] = dp[i+1][j-1] + dp[i-1][j-1] + dp[i][j-1];
                // cout<<i<<" "<<j<<" "<<dp[i][j]<<endl;
            }
        }
        return dp[0][steps]%MOD;
    }
};