问题 H: 吃西瓜
时间限制: 1 Sec 内存限制: 256 MB
提交: 8 解决: 6
[提交][状态][讨论版][命题人:外部导入]
题目描述
【问题描述】
老胡买了是长方体形的西瓜来犒劳大家....
这块西瓜长m厘米,宽n厘米,高h厘米.他发现如果把这块西瓜平均地分成m*n*h块1立方厘米的小正方体,那么每一小块都会有一个营养值(可能为负,因为西瓜是有可能坏掉的,但是绝对值不超过200)。
现在老胡决定从这m*n*h立方厘米的西瓜中切出mm*nn*hh(0<=mm<=m,0<=nn<=n,0<=hh<=h)立方厘米的一块小西瓜(一定是立方体形,长宽高均为整数),送给该场比赛最高分获得者补充营养。他想知道他最多能获得多少营养值。
换句话说,我们希望从一个m*n*h的三维矩阵中,找出一个三维子矩阵,这个子矩阵的权和最大.
一个2*3*4的例子,最优方案为切红色2*3*1部分。
【文件输入】matrix.in
首行三个正整数h,m,n(注意顺序),分别表示西瓜的高,长,宽.
以下h部分,每部分是一个m*n的矩阵,第i部分第j行的第k个数表示西瓜第i层,第j行第k列的那块1立方厘米的小正方体的营养值.
【文件输出】matrix.out
老胡所能得到的最大营养值
【输入输出样例】
matrix.in | matrix.out |
2 3 4 4 1 2 8 0 5 -48 4 3 0 1 9 2 1 4 9 1 0 1 7 3 1 2 8 | 45 |
【数据规模】
对于30%的数据,h=1,1<=m,n<=10
对于全部的数据,1<=h<=32,1<=m,n<=50,保证h<=m,n
[说明]此题中出现的所有数全为整数。
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int inf = 2e9;
int h, m, n;
int a[55][55][55];
int qian[55], mi[55];
int cal(int t[])
{
int ans = -inf;
qian[0] = 0;
for (int i = 1; i <= h; i++)
{
qian[i] = qian[i - 1] + t[i];
mi[i] = min(mi[i - 1], qian[i]);
ans = max(ans, qian[i] - mi[i - 1]);
}
return ans;
}
int t[55];
int sum[55][55][55];
int main()
{
cin >> h >> m >> n;
for (int i = 1; i <= h; i++)
for (int j = 1; j <= m; j++)
for (int k = 1; k <= n; k++)
cin >> a[i][j][k];
for (int i = 1; i <= h; i++)
{
for (int j = 1; j <= m; j++)
for (int k = 1; k <= n; k++)
sum[i][j][k] = sum[i][j - 1][k] + sum[i][j][k - 1] - sum[i][j - 1][k - 1] + a[i][j][k];
}
int ans = 0;
for (int i = 1; i <= m; i++) // 枚举四边界
for (int j = i; j <= m; j++)
for (int p = 1; p <= n; p++)
for (int q = p; q <= n; q++)
{
for (int hh = 1; hh <= h; hh++) // 将取出的方块每一层加成一个数
t[hh] = sum[hh][j][q] - sum[hh][j][p - 1] - sum[hh][i - 1][q] + sum[hh][i - 1][p - 1];
ans = max(ans, cal(t)); // 在新得到的串中 O(n) 取和最大的一段
}
cout << ans << endl;
return 0;
}