LeetCode题解——停在原地的方案数

LeetCode题解——停在原地的方案数

• 题目介绍

• 解题思路

1. 动态规划，将整个大题进行拆分，声明一个容器dp[n][m],n代表多少步，m是数组下标
2. 已知要走到dp[n][0]，有2种方法，dp[n-1][0]保持不变，dp[n-1][1]左移一步
3. 因此走到dp[n][0]的方法就是走到dp[n-1][0]的方法数加上走到dp[n-1][1]的方法数
4. 依次类推dp[n-1][1] = dp[n-2][0] + dp[n-2][1] + dp[n-2][2]，就这样一直算到dp[2][0] = dp[1][0] + dp[1][1]
5. dp[1][0] = dp[1][1] = dp[0][0] = 1

class Solution {
public:
    const int MODULO = 1000000007;

    int numWays(int steps, int arrLen) {
        int maxColumn = min(arrLen - 1, steps);
        vector<vector<int>> dp(steps + 1, vector<int>(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];
    }
};