LeetCode 3665. 统计镜子反射路径数目

很经典的题型，但由于有方向限制，在更新 dp 时需要多考虑方向。

定义：dpij0dpij0dpij0 表示从上方到达 (i,j)(i, j)(i,j)的路径数，dpij1dpij1dpij1 表示从左方到达 (i,j)(i, j)(i,j)的路径数。

则到达某个格子的总路径数就是 dpij0+dpij1dpij0 + dpij1dpij0+dpij1。

更新时，分方向讨论。

若上方为空格，则：dpij0=dpi−1j0+dpi−1j1dpij0=dpi-1j0+dpi-1j1dpij0=dpi−1j0+dpi−1j1 ；

若左方为空格，则：dpij1=dpij−10+dpij−11dpij1=dpij-10+dpij-11dpij1=dpij−10+dpij−11

若上方为镜子，则：dpij0=dpi−1j1dpij0=dpi-1j1dpij0=dpi−1j1 ；

若左方为镜子，则：dpij1=dpij−10dpij1=dpij-10dpij1=dpij−10

代码如下：

#define MOD 1000000007
class Solution {
public:
    int uniquePaths(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        int ans = 0;
        vector dp(m, vector<vector<long long int> >(n, vector<long long int>(2, 0)));
        dp[0][0][0] = 1;
        for(int i = 0;i<m;i++){
            for(int j = 0;j<n;j++){
                if(i==0 && j==0) continue;
                if(i > 0){
                    if(grid[i-1][j]==0){
                        dp[i][j][0] = (dp[i-1][j][0] + dp[i-1][j][1])%MOD;
                    } else{
                        dp[i][j][0] = dp[i-1][j][1]%MOD;
                    }
                }
                if(j>0){
                    if(grid[i][j-1]==0){
                        dp[i][j][1] = (dp[i][j-1][1] + dp[i][j-1][0])%MOD;
                    } else{
                        dp[i][j][1] = dp[i][j-1][0]%MOD;
                    }
                }
            }
        }
        if(grid[m-1][n-1] == 1){
            return 0;
        } else{
            return (dp[m-1][n-1][0] + dp[m-1][n-1][1]) %MOD;
        }
    }
};