題意: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';
}