CSES-1638 Grid Paths

题目描述

原题链接

考虑一个n × n方格可能有陷阱的网格。不允许移动到有陷阱的方格。
您的任务是计算从左上角方块到右下方块的路径数。您只能向右或向下移动。

算法描述

状态定义：

• dp[i][j]：表示走到grid[i][j]的路径数。

状态转移：

• 若grid[i][j] == '*'，则说明不可达，dp[i][j] = 0。
• 若grid[i][j] == '.'，则可能从左边的格子和上面的格子到达，对应的状态转移方程为dp[i][j] = dp[i-1][j] + dp[i][j-1]

初始状态dp[0][0]：取决于grid[i][j]是否为'*'。

边界情况：

• 上边界：状态转移方程取决于左边的格子。
• 左边界：状态转移方程取决于上边的格子。

状态压缩

观察状态转移方程，可以发现第 i 行第 j 列的格子只取决于第 i-1 行第 j 列的格子和第 i 行第 j-1 列的格子，因此我们可以将二维的dp[i][j]压缩为dp[j]。状态转移过程中，dp[0...j-1]表示第 i 行的 0 到 j-1 的格子，dp[j...n-1]表示第 i-1 行的 j 到 n-1 的格子。

复杂度分析

时间复杂度：$O(n^2)$
空间复杂度：$O(n)$

程序代码

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010;
const int MOD = 1e9 + 7;
char grid[N][N];
int dp[N];

int main()
{
    int n;
    cin >> n;
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            cin >> grid[i][j];
        }
    }
    int res = 0;
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            if(i == 0 && j == 0) {
                dp[j] = grid[i][j] == '*' ? 0 : 1;
            }
            else if(i == 0) {
                dp[j] = grid[i][j] == '*' ? 0 : dp[j-1];
            }
            else if(j == 0) {
                dp[j] = grid[i][j] == '*' ? 0 : dp[j];
            }
            else {
                dp[j] = grid[i][j] == '*' ? 0 : (dp[j] + dp[j-1]) % MOD;
            }
        }
    }
    cout << dp[n-1] << endl;
    return 0;
}