洛谷[P1719 最大加权矩形] {前缀和与差分}
题目描述
为了更好的备战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列的矩阵。
输出格式
最大矩形(子矩阵)的和。
输入输出样例
输入
4
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
输出
15
说明/提示
n<=120
解题思路
首先,使用**b[i][j]来表示矩阵i*j覆盖下的价值和。b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j],求解b[i][j]如图:
因为b[i-1][j]+b[i][j-1]多加了b[i-1][j-1],所以要减去,而且还要加上当前位置的值a[i][j]。
因为b[i-1][j]+b[i][j-1]多加了b[i-1][j-1],所以要减去,而且还要加上当前位置的值a[i][j]。
然后,求解矩阵覆盖下的价值和,等于b[i][j]-b[i-p][j]-b[i][j-q]+b[i-p][j-q](p、q分别代表矩形的宽(y方向)与长(x方向))。
表达式解释由图所示:(p=3,q=2)
因为-b[i-p][j]-b[i][j-q]相当于减去两次的b[i-p][j-q],所以需要加回来一次。(红色边框即为所求矩阵)
完整代码
#include<bits/stdc++.h>
using namespace std;
int a[300][300],b[300][300];
int main()
{
int N;
scanf("%d",&N);
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
{
scanf("%d",&a[i][j]);
b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j];//b[i][j]表示i*j的矩阵覆盖下的总权值和
}
int ans=-1e9;//最初让ans尽可能地小
for(int i=1;i<=N;i++)
{
for(int j=1;j<=N;j++)
{
for(int p=1;p<=i;p++)//保证p>=i,确保p-i>=0
{
for(int q=1;q<=j;q++) //保证q>=j,确保q-j>=0
{
ans=max(ans,b[i][j]-b[i-p][j]-b[i][j-q]+b[i-p][j-q]);//计算子矩阵覆盖下的权值和
}
}
}
}
printf("%d",ans);
return 0;
}