3-6 奥马尔·爱糖果 (30分)
奥马尔喜欢吃很多糖果,但不幸的是,大部分的糖果都不健康。所以他的父母找到了一种方法,给每个糖果打分,分数越高,糖果就越健康(分数是一个整数,可以是正的、零的或负的)。一个孩子和他的父母买了一些糖果,他们到了卖糖果的地方,所有的糖果都被存储在一个N*M的二维网格中,每一行有M个糖果。行从上到下从1到N编号,列从左到右从1到M编号,每个单元格包含一个糖果健康值。奇怪的是,这个二维网格中的糖果的健康值总是从上到下、从左到右依次增大
-4 | -2 | -1 |
---|---|---|
-3 | 2 | 3 |
1 | 4 | 5 |
奥马尔想买完一个子矩阵中的所有糖果,请你编程序求出他所能买得的最大的糖果健康值。
输入格式:
输入的第一行是单个整数T,测试用例的数量(1≤T≤100)。每个测试用例开始一行包含空格隔开两个整数N M (1≤N,M≤1000)代表糖果网格的尺寸,紧随其后的是N行,每一个包含M个空格隔开整数,代表这一行糖果的健康值。网格中的每个整数将不小于- 2000,也不大于2000。
输出格式:
对于每个测试用例,打印包含单个整数的一行,该整数表示从非空子矩形中可以获得的健康值的最大总和。
输入样例:
在这里给出一组输入。例如:
1
3 3
-4 -2 -1
-3 2 3
1 4 5
输出样例:
在这里给出相应的输出。例如:
14
思路
-4 | -2 | -1 |
---|---|---|
-3 | 2 | 3 |
1 | 4 | 5 |
可以从右下角开始求以其为左上角的子矩阵的值,值最大时取此子矩阵及值;
以i为行 j为列,f[i][j]为子矩阵值,a[i][j]为该格糖果的值,i=末行时,f[i][j]=a[i][j]+f[i+1][j];j==末列时,f[i][j]=a[i][j]+f[i][j+1];否则f[i][j]=f[i][j]时+f[i+1][j]值为f[i + 1][j] + f[i][j + 1] + a[i][j] - f[i + 1][j + 1]。
。。。 | -2+(-1+3+5)+(3+5+4+5-5)-(3+5) | -1+3+5 |
---|---|---|
。。。 | 2+(3+5)+(3+5)-5 | 3+5 |
1+4+5 | 4+5 | 5 |
C++代码
#include<iostream>
#include<algorithm>
#include<limits.h>
#include<string.h>
#pragma GCC optimize (2)
#pragma GCC optimize (3)
using namespace std;
int a[1005][1005], f[1005][1005];
int main() {
int T, m, n;
scanf("%d", &T);
while (T--) {
int sum = INT_MIN;
scanf("%d %d", &n, &m);
memset(a, -20000, sizeof(a));
memset(f, 0, sizeof(f));
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
scanf("%d", &a[i][j]);
for (int i = n; i >= 1; i--) {
int t;
for (int j = m; j >= 1; j--) {
f[i][j] = a[i][j];
t = j;
if (i == n)
f[i][j] = max(f[i][j + 1] + a[i][j], a[i][j]);
else if (j == m)
f[i][j] = max(f[i + 1][j] + a[i][j], a[i][j]);
else
f[i][j] = max(a[i][j], f[i + 1][j] + f[i][j + 1] + a[i][j] - f[i + 1][j + 1]);
sum = max(sum, f[i][t]);
}
sum = max(sum, f[i][t]);
}
printf("%d\n", sum);
}
return 0;
}