上链接:https://www.luogu.com.cn/problem/P1719#submit
上题干:
题目描述
为了更好的备战 NOIP2013,电脑组的几个女孩子 LYQ,ZSC,ZHQ 认为,我们不光需要机房,我们还需要运动,于是就决定找校长申请一块电脑组的课余运动场地,听说她们都是电脑组的高手,校长没有马上答应他们,而是先给她们出了一道数学题,并且告诉她们:你们能获得的运动场地的面积就是你们能找到的这个最大的数字。
校长先给他们一个 n×n 矩阵。要求矩阵中最大加权矩形,即矩阵的每一个元素都有一权值,权值定义在整数集上。从中找一矩形,矩形大小无限制,是其中包含的所有元素的和最大 。矩阵的每个元素属于[−127,127] ,
例如
0 –2 –7 0 9 2 –6 2 -4 1 –4 1 -1 8 0 –2
在左下角:
9 2 -4 1 -1 8
和为 15。
几个女孩子有点犯难了,于是就找到了电脑组精打细算的 HZH,TZY 小朋友帮忙计算,但是遗憾的是他们的答案都不一样,涉及土地的事情我们可不能含糊,你能帮忙计算出校长所给的矩形中加权和最大的矩形吗?
输入格式
第一行:n,接下来是 n 行 n 列的矩阵。
输出格式
最大矩形(子矩阵)的和。
输入输出样例
输入 #1复制
4 0 -2 -7 0 9 2 -6 2 -4 1 -4 1 -1 8 0 -2输出 #1复制
15说明/提示
1≤n≤120
这是一道二维数组的前缀和模板题,我们可以用画图来理解:
我们先从答案开始往前递推:
假设我们需要求任意一个子矩阵的矩阵和:设(x1,y1)为左上角,(x2,y2)为右下角
我们利用一维前缀和的思想:求(x,y)区间的字段和 为 S【y】- S【x】
可以把这个小矩形,变成几个矩形的差值。
设S【x,y】为以【x,y】为右下角的矩阵和
那么则有
以(x1,y1)为左上角,(x2,y2)为右下角的矩阵和 =S【x2,y2】+S【x1-1,y1-1】-S【x2,y1-1】-S【x1-1,y2】
这是S【x2,y2】+S【x1-1,y1-1】(绿色部分为双重覆盖
这是 S【x2,y2】+S【x1-1,y1-1】-S【x2,y1-1】
这是S【x2,y2】+S【x1-1,y1-1】-S【x2,y1-1】-S【x1-1,y2】
好了,现在我们要解决的就是怎么用S【x,y】表示【x,y】为右下角的矩阵和
我们还是用同样的思想,将S【x,y】分成多个矩阵和差
S【x,y】=S【x-1,y】+S【x-1,y】-S【x-1,y-1】+a【x,y】;
这是S【x-1,y】+S【x-1,y】 绿色部分为重叠部分
上代码:
const int N = 130;
int a[N][N];
int s[N][N];
int sum = -0x7fffffff;
int qsum(int x1, int y1, int x2, int y2)
{
return s[x2][y2] + s[x1 - 1][y1 - 1] - s[x2][y1 - 1] - s[x1 - 1][y2];
}
int main()
{
int n;
cin >> n;
for(int i=1;i<=n;i++)
for (int j = 1; j <= n; j++)
{
cin >> a[i][j];
s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
}
for(int x1=1;x1<=n;x1++)
for(int y1=1;y1<=n;y1++)
for(int x2=x1;x2<=n;x2++)
for (int y2 = y1; y2 <= n; y2++)
{
sum = max(sum, qsum(x1, y1, x2, y2));
}
cout << sum;
}