[蓝桥杯试题] 算法训练 拿金币 C++解法

资源限制
内存限制：256.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s
问题描述
　　有一个N x N的方格,每一个格子都有一些金币,只要站在格子里就能拿到里面的金币。你站在最左上角的格子里,每次可以从一个格子走到它右边或下边的格子里。请问如何走才能拿到最多的金币。
输入格式
　　第一行输入一个正整数n。
　　以下n行描述该方格。金币数保证是不超过1000的正整数。
输出格式
　　最多能拿金币数量。
样例输入
3
1 3 3
2 2 2
3 1 2
样例输出
11
数据规模和约定
　　n<=1000

• 想要走到{i, j}(不包含边界的情况),只有两种走法,要么从左边过来,要么从上面下来,所以很容易写处状态转移方程
  👉dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]) + nums[i][j]
  对于处于边界的点,我们令其附近处于边界之外的点的dp值为0,这样就不会走出边界了

代码

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 1010;
int nums[N][N], dp[N][N];
int n;

int main(int argc, char *argv[]) {
    cin >> n;
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j)
        scanf("%d", &nums[i][j]);
    
    for (int i = 1; i <= n; ++i) 
        for (int j = 1; j <= n; ++j) 
            dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]) + nums[i][j];

    cout << dp[n][n];
    return 0;
}

优化成一维dp数组的写法👇

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 1010;
int nums[N][N], dp[N];
int n;

int main(int argc, char *argv[]) {
    cin >> n;
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j) {
        // 当前节点的计算只涉及前一个节点和上一个节点,故可以边输入边计算
        	scanf("%d", &nums[i][j]);
        	dp[j] = max(dp[j], dp[j - 1]) + nums[i][j];
        }
    
    cout << dp[n];
    return 0;
}

个人认为最简代码

不过还是希望大家不要写最简的代码,代码简短和代码无注释都不利于理解!

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 1010;
int nums[N], dp[N];
int n;

int main(int argc, char *argv[]) {
    cin >> n;
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j) {
        	scanf("%d", &nums[j]);
        	dp[j] = max(dp[j], dp[j - 1]) + nums[j];
        }
    
    cout << dp[n];
    return 0;
}