洛谷 P1004 [NOIP2000 提高組] 方格取數

_Yxc發表於2024-04-04

題意:n*n的方格,從左上角到右下角兩次。每一次經過的路徑中,如果有數字,數字都會變成0並計數。求兩次路徑的最大計數。

思路:線性dp,從左上角到右下角步數固定為 2 * n - 2步。 初始時0步dp[0][1][1] = grid[1][1],知道了x1和x2可以確定對應的y,可以直接進行狀態轉移。 可以增加剪枝:x <= min(n, s1 + 1)。

總結:
1 忽略了當到了終點時x1 == x2的情況,被直接跳過了。
2 忽略了x不動,當前步數都是由y移動的情況(4種轉移狀態之一)

  void solve(){
    int n;
    cin >> n;

    vector<vector<int>> grid(n + 1, vector<int>(n + 1, 0));
    {
        int x, y, k;
        while (cin >> x >> y >> k && (x && y && k)){
            grid[x][y] = k;
        }
    }

    vector<vector<vector<int>>> dp(2, vector<vector<int>>(n + 1, vector<int>(n + 1)));
    dp[0][1][1] = grid[1][1];
    int p = 0;
    for (int s = 1; s <= 2 * n - 2; ++s){
        p ^= 1;
        for (int x1 = 1; x1 <= min(n, s + 1); ++x1){
                for (int x2 = 1; x2 <= min(n, s + 1); ++x2){
                    if ((x1 != x2) || s == 2 * n - 2){
                        int y1 = s + 2 - x1;
                        int y2 = s + 2 - x2;
                        if (y1 <= n && y2 <= n){
                            auto& cur = dp[p][x1][x2];
                            if (x1 > 1 && x2 > 1){
                                cur = max(cur, dp[p ^ 1][x1 - 1][x2 - 1] + grid[x1][y1] + grid[x2][y2]);
                            }
                            if (x1 > 1 && y2 > 1){
                                cur = max(cur, dp[p ^ 1][x1 - 1][x2] + grid[x1][y1] + grid[x2][y2]);
                            }
                            if (x2 > 1 && y1 > 1){
                                cur = max(cur, dp[p ^ 1][x1][x2 - 1] + grid[x1][y1] + grid[x2][y2]);
                            }
                            if (y1 > 1 && y2 > 1){
                                cur = max(cur, dp[p ^ 1][x1][x2] + grid[x1][y1] + grid[x2][y2]);
                            }
                        }
                    }
                }
        }
    }/*
    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= n; ++j){
            cout <<  grid[i][j] << " \n"[j == n];
        }
    }
    for (int i = 1; i <= n; ++i){
        for (int j = 1; j <= n; ++j){
            cout <<  dp[p][i][j] << " \n"[j == n];
        }
    }*/
    cout << dp[p][n][n] - grid[n][n] << '\n';

}

相關文章