P1004 [NOIP2000 提高组] 方格取数

 复制Markdown  展开

题目背景

NOIP 2000 提高组 T4

题目描述

设有 N×NN×N 的方格图 (N≤9)(N≤9)，我们将其中的某些方格中填入正整数，而其他的方格中则放入数字 00。如下图所示（见样例）:

某人从图的左上角的 AA 点出发，可以向下行走，也可以向右走，直到到达右下角的 BB 点。在走过的路上，他可以取走方格中的数（取走后的方格中将变为数字 00）。
此人从 AA 点到 BB 点共走两次，试找出 22 条这样的路径，使得取得的数之和为最大。

输入格式

输入的第一行为一个整数 NN（表示 N×NN×N 的方格图），接下来的每行有三个整数，前两个表示位置，第三个数为该位置上所放的数。一行单独的 00 表示输入结束。

输出格式

只需输出一个整数，表示 22 条路径上取得的最大的和。

输入输出样例

输入 #1复制

8
2 3 13
2 6  6
3 5  7
4 4 14
5 2 21
5 6  4
6 3 15
7 2 14
0 0  0

输出 #1复制

67

说明/提示

数据范围：1≤N≤91≤N≤9。

代码

点个赞再走吧

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
    int N;
    cin >> N;
    vector<vector<int>> grid(N, vector<int>(N, 0));
    while (true) {
        int x, y, val;
        cin >> x >> y >> val;
        if (x == 0 && y == 0 && val == 0) {
            break;
        }
        grid[x-1][y-1] = val;
    }
    
    vector<vector<int>> dp(N, vector<int>(N, 0));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (i == 0 && j == 0) {
                dp[i][j] = grid[i][j];
            } else if (i == 0) {
                dp[i][j] = dp[i][j-1] + grid[i][j];
            } else if (j == 0) {
                dp[i][j] = dp[i-1][j] + grid[i][j];
            } else {
                dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + grid[i][j];
            }
        }
    }
    
    int ans = dp[N-1][N-1];
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (i == 0 && j == 0) {
                dp[i][j] = grid[i][j];
            } else if (i == 0) {
                dp[i][j] = dp[i][j-1] + grid[i][j];
            } else if (j == 0) {
                dp[i][j] = dp[i-1][j] + grid[i][j];
            } else {
                dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + grid[i][j];
            }
        }
    }
    
    ans = max(ans, dp[N-1][N-1]);
    cout << ans << endl;
    
    return 0;
}